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,自己又踩了不少坑。
最重要的一点是要搞清楚各个包的对应版本,然后可以手动下载镜像。

PyTorch镜像下载地址

cuda支持版本:PyTorch环境搭建:Win10+CUDA9.2+CUDNN配置pytorch(GPU)_第1张图片

torch和torchvision版本对应关系:PyTorch环境搭建:Win10+CUDA9.2+CUDNN配置pytorch(GPU)_第2张图片

参考原文地址

遇到的最大的坑

我一开始按照第一篇博文是成功的,后来想安装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~
不过好在搞清楚了配置环境的一些重要事项)

PyTorch环境搭建:Win10+CUDA9.2+CUDNN配置pytorch(GPU)_第3张图片
在这里插入图片描述

你可能感兴趣的:(PyTorch)