Windows10+CUDA9.2+CUDNN配置pytorch(花了两天时间走了不少坑,即使现在已经晚上两点过了,但还是把这两天的总结一下)
环境说明
-
Windows10
-
Python3.6.5
-
CUDA9.2.217
-
pycharm
最后安装的版本 -
anaconda3
-
cuda_9.2.148
-
cudnn-9.2-windows-7.2.138
步骤
因为之前已经配置好了python的环境(这里不说明安装配置python环境等):
-
anaconda3安装(这个很简单,选择对应的Python环境就可以)
需要注意的是,环境变量的配置问题以及添加国内镜像
我配置了这些,参考别人的解决办法(有的只配置了两个)
最简单的是在安装anaconda的时候默认勾选加入环境变量(建议还是使用自定义)
配置完成之后,可以在命令行输入conda --version来检查是否成功,或者命令行输入python也可以 -
选择电脑适合的CUDA进行安装
首先要查看自己的电脑有没有独立显卡(在计算机管理-设备管理器-显示适配器中可以查看)
同时一定要自己去看本机适合的CUDA版本(我之前就是版本弄错了,一直没有注意到,浪费了很多时间,同时如果不使用多个cuda,最好卸载掉下错的那个以及删除配置的环境变量)。打开NVIDIA的控制面板,在菜单栏的帮助一栏中查看系统信息-显示组件:我的就是CUDA9.2.217
找到了自己合适的版本之后,就可以去官网下载相应的CUDA了(点击此处)
接下来就是一系列next的安装步骤了(安装anaconda3的位置我没有改,默认系统安装,方便以后查找)
最后,检查是否成功:命令行输入nvcc -V,显示了版本信息则正确。 -
安装相应的CUDNN
安装CUDNN必须和CUDA的版本保持一致,下载的时候很慢,也有可能下载不成功,这里的方法是点击右键复制链接,使用迅雷下载。下载成功解压之后将CUDNN目录下的lib,bin,include复制到CUDA\v9.2目录下,替换掉原来的文件。
-
安装pytorch
官网下载地址
不知道为什么Python3.6+CUDA9.2不配拥有直接的下载链接了,使用命令行下载也只会报错:在镜像中找不到对应的包。没办法!又看见了别人使用的一个办法:离线下载.whl文件(不过需要自己知道自己需要下载的版本)
同样下载的时候使用复制链接到迅雷下载会相对快很多,也不会下载失败。我的版本是
8.离线安装.whl文件
命令行进入到存放上述两个whl文件的目录,使用pip来安装(安装之前可以先查看一下是否安装了wheel,没有的话需要安装:pip install wheel)直接命令行输入:pip install whl’s name.whl(一定要先安装torch,然后再安装torchvision).
最后,查看是否成功:点击开始-打开anaconda prompt:
遇到的错误以及参考链接
- 安装anaconda:https://www.zhihu.com/question/58033789/answer/254673663
- 清华大学开源镜像网:https://mirrors.tuna.tsinghua.edu.cn/
- 检查显卡支持的cuda版本:https://jingyan.baidu.com/article/6fb756ec4fabc4241858fbf7.html
- 参考安装过程:https://blog.csdn.net/Mind_programmonkey/article/details/99688839/(这位作者写的很详细,由于后面版本不一样,也遇到了许多不同的问题)