pytorch + cuda + cudnn 配置完整教程

①查看对应版本并下载（下载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.csdn.net/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.csdn.net/fffsolomon/article/details/107593407
 
 

②卸载之前的版本
参考这篇博客：https://blog.csdn.net/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.csdn.net/sinat_23619409/article/details/84202651
也可以参考这两篇
https://blog.csdn.net/jhsignal/article/details/111401628
下面这篇博客中提到的环境变量最好添加一下：
https://blog.csdn.net/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