①查看对应版本并下载(下载python、cuda、cudnn和visual studio)
首先查看python、torch、cuda、cudnn和Visual Studio对应的版本,注意版本最好严格对应。
具体对应关系看下面这篇博客和torch官网。cuda对应的Visual Studio版本可以百度搜索,如果Visual Studio版本不对cuda在安装时会进行警告。
https://pytorch.org/get-started/previous-versions/
https://blog.youkuaiyun.com/kellyroslyn/article/details/109668001
如果需要创建python3.5、3.6、3.7等等其它环境,可以直接在Anaconda Prompt(管理员打开) 中输入对应指令:
conda create -n python35 python=3.5 anaconda
conda create -n python36 python=3.6 anaconda
…
两句代码会安装所有相关的包。
如果只需要相关环境,可以执行:
conda create -n python35 python=3.5
conda create -n python36 python=3.6
…
安装完成后可以直接在开始菜单看到对应版本的spyder、控制台等等。
具体可以参考这个:
https://blog.youkuaiyun.com/fffsolomon/article/details/107593407
②卸载之前的版本
参考这篇博客:https://blog.youkuaiyun.com/shuiyuejihua/article/details/78738664
在windows的软件管理里,这两个不卸载:NVIDIA的图形驱动程序、NVIDIA Physx系统软件。除了这两个之外的关于cuda的都需要卸载,可以查看软件的安装时间,和cuda同一天安装的都需要卸载,卸载完成后:
1.最好再清理一下注册表;
2.手动删除C:\Program Files\NVIDIA GPU Computing Toolkit文件;
3.在环境变量里把与当前cuda有关的都删除掉。
③安装cuda和cudnn
参考这篇博客,严格按照步骤来。不同版本的cuda和cudnn安装大同小异。https://blog.youkuaiyun.com/sinat_23619409/article/details/84202651
也可以参考这两篇
https://blog.youkuaiyun.com/jhsignal/article/details/111401628
下面这篇博客中提到的环境变量最好添加一下:
https://blog.youkuaiyun.com/jer8888/article/details/100558964
③安装torch
按照官网查看对应版本的指令,在pip中进行安装。
https://pytorch.org/get-started/previous-versions/
另外如果遇到
ERROR: torch has an invalid wheel, .dist-info directory not found
可以参考这篇博客
https://www.cnblogs.com/jaysonteng/p/15183610.html