Torch官网的版本只提供了CUDA 9.2和CUDA 10.1的版本,我的cuda是10.0的。所以这里版本不对应导致Torch.cuda加速无法运行。
step1:删除原版本CUDA
step2:删除原版本PyTorch
conda uninstall pytorch
step3:重新安装新10.1版本的CUDA和PyTorch
使用语句
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
验证环节报错1:
使用语句:
conda install pytorch== 1.6.0 torchvision==0.7.0 cudatoolkit=10.1
安装环节报错2: