核心的步骤按这篇文章:
一些可能出错的地方解决方法:
cuda的安装最好是默认地址不要改
下载pytorch时最好不要用清华源,使用本地下载最好(附地址如下)
清华源应该是只能映射CPU版本,想要下载GPU版本不能使用清华源,
所以可以下载到本地然后再安装
一个是pytorch一个是torchversion,两个都下
download.pytorch.org/whl/torchvision/
download.pytorch.org/whl/torch/
具体实现本地安装方法见此文:
安装pytorch(下载到本地进行简单安装)_pytorch本地下载-优快云博客
查看自己电脑应该装哪个版本cuda和pytorch看这篇: