其他博客的方法,要么傻要么有坑要么不全。
特别是对应cuda和pytorch版本的方法,忙活半天,最后发现没有一个人用对方法了。
真相是,你想安装pytorch,最好就别让pip自己去下载安装,而是你找好wheel文件、下载到本地后再pip安装。
--------------------------------------------------分割线--------------------------------------------------
我的环境:NVIDIA独显,win10
右击桌面,进入NVIDIA控制面板
系统信息-组件,可以看到我的显卡最高支持CUDA10.1。我可以装CUDA9.0、9.2,但不能装10.2
我们要下载的是这两个东西:
大陆访问NVIDIA网站贼慢,甚至可能打不开,所以最好有梯子。
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
选择自己要的版本下载即可
注意:
CUDA一定要选择“exe[local]”!!!在线安装包是巨坑,2G多的安装文件下载个半天,一旦安装结束就删除,如果安装失败了,那你重新安装时还得再慢慢下载一遍。
根据自己的CUDA版本下载对应的cudnn,下载地址:https://developer.nvidia.com/rdp/cudnn-archive
尤其最好先别装Visual Studio Integration。因为如果安装失败基本都是它导致的。
如果安装Visual Studio Integration失败,参考这篇博客:
1. CUDA安装失败解决方法_zzpong的小屋-CSDN博客_cuda安装失败
其他博客那些方法,让你conda/pip install 一个官网生成的命令,可供选择的版本少还贼不可靠,甭管它们。
其实只要浏览器访问这个地址:
https://download.pytorch.org/whl/cu+你的版本号
例如cuda11.3,那就是访问https://download.pytorch.org/whl/cu113
我是cuda10.1,那就是访问https://download.pytorch.org/whl/cu101
这里需要注意:
1,我们是已知CUDA版本来找pytorch的whl文件,别找到CPU版去了
我下载的这个文件名:torch-1.8.1+cu101-cp38-cp38-win_amd64.whl
意思是pytorch版本为1.8.1,CUDA版本为10.1,python版本为3.8,windows64位
2,如果是安装torchvision,它的版本要和torch版本匹配!匹配关系如下表
(该表来自https://github.com/pytorch/vision#installation)
这里是个坑,这些whl居然必须改名才能安装,也是服了。
需要把文件名末尾“win_amd64”改成“win32”:
既然是win10,就别守旧还老用cmd命令行了。
右击在whl文件所在目录的空白处 ,打开Powershell:
pip install 该whl,成功: