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
来检查是否成功,或者命令