1. 首页
  2. 移动开发
  3. 其他
  4. PyTorch build:提供适用于各种环境的PyTorch的Docker构建序列 源码

PyTorch build:提供适用于各种环境的PyTorch的Docker构建序列 源码

上传者: 2021-04-06 23:37:49上传 ZIP文件 5.07KB 热度 8次
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
用户评论