PyTorch环境搭建:Win10+CUDA9.2+CUDNN配置pytorch(GPU)
Win10+CUDA9.2+CUDNN配置pytorch(GPU)遇到的坑
试了很多博客的方法都没有成功,最终成功主要参照是这篇博客:Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑),但是我安装的是torch1.3.0+torchvision0.4.1,自己又踩了不少坑。
最重要的一点是要搞清楚各个包的对应版本,然后可以手动下载镜像。
cuda支持版本:
torch和torchvision版本对应关系:
遇到的最大的坑
我一开始按照第一篇博文是成功的,后来想安装pytorch1.x,按照对应版本下载重装了,对应关系是对的,但是就是失败,出现了各种错误,anaconda卸了重装好几次。。
最后发现我在配置好之后每次用的是py命令进入python环境的,而一开始是使用python是可以的。我惊奇的发现键入python和py竟然python版本是不一样的,最后发现base环境和pytorch环境中的python版本是不一样的,由于安装时勾选了add to path,在命令行中输入python直接用的是base环境中的python版本,而新建环境中python版本需要进入该环境才能使用(即使进入该环境下,使用py命令也是base环境的python版本,应该使用python命令进入才是正确的python版本)。
(之前配置的TensorFlow又要重新配了。555~
不过好在搞清楚了配置环境的一些重要事项)