Pytorch多机多卡分布式训练

2021-01-08 15:14 1971182

Pytorch 分布式环境搭建 当前云轩服务器环境：（已经可用分布式） cuda_10.0.130 cudnn_7.6.5.32 pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 卸载 CUDA ： sudo /usr/local/cuda-8.0/bin/uninstall_cuda_.pl sudo rm -rf /usr/local/cuda- 安装 CUDA ： https://developer.nvidia.com/cuda-toolkit...