PyTorch build:提供适用于各种环境的PyTorch的Docker构建序列 源码
PyTorch构建 提供适用于各种环境的Docker PyTorch构建序列。 1. Docker映像环境 Ubuntu 20.04 x86_64 CUDA 11.2 cuDNN 8.1 PyTorch v1.8.1(从源代码构建。它将在Docker构建过程中自动下载。) TorchVision v0.9.1 TorchAudio v0.8.1 2.默认构建参数 -- -- ******** Summary ******** -- General: -- CMake version : 3.18.4 -- CMake command : /usr/local/lib/python3.8/dist-packages/cmake/data/bin/cmake -- System : Linux -- C
用户评论