在Win10上手动安装PyTorch的流程(不使用Anaconda,2020年8月)

上一篇博客中,我总结了使用Anaconda安装PyTorch的过程,不得不说这个过程是简便的,但是我还是觉得Anaconda有没处理好的地方(例如我吐槽的目录位置),另外,我在Anaconda的Prompt中输入import torch是正常的,但是在VS Code中运行,就会报

File "C:\Users\%我的用户名%\.conda\envs\pytorch\lib\site-packages\torch\__init__.py", line 189, in <module>
    from torch._C import *
ImportError: numpy.core.multiarray failed to import

这样的错误,看看就知道是Anaconda安装numpy时候的问题。也懒得花时间去解决了,准备再试一下不用Anaconda时候的流程。

首先,我们在这里:https://developer.nvidia.com/cuda-downloads,手动下载CUDA。

需要注意的是,通过https://pytorch.org/get-started/locally/,我们可以看到,PyTorch目前支持的CUDA版本是9.2,10.1,10.2,保险起见就安装10.2版本的CUDA。所以我们应该在NVIDIA的那个页面中选择“Legacy Releases”——选择CUDA 10.2的链接,然后点选几下即可(为啥会有2.4GB这么大)。

按照这里的介绍:https://www.jianshu.com/p/d7851ffea38b,需要在安装CUDA之前安装Visual Studio,这点我实在太不理解了,但是尝试安装CUDA的时候确实提示说要先装Visual Studio,我们在这里选择Community版:https://visualstudio.microsoft.com/zh-hans/,参考上面网页中的介绍,再安装CUDA即可。

安装完后,在cmd中输入nvcc -V,如果能看到10.2,则说明安装成功。

下面再下载cuDNN,注册号账号之后在这里:

https://developer.nvidia.com/rdp/cudnn-download

选择对应的cudnn版本进行下载。然后还是按照上面的网页拷贝几个文件夹。

这样貌似就可以通过pip安装了,但是我竟然在这一步也踩到坑了,不得不说,Python的整个Eco-system太混乱了!

由于是在一台新的机子上安装,在这里https://www.python.org/downloads/,我就直接点了Download Python 3.8.5,这里另外吐槽一下:为啥默认下载的是32位的??检测一下操作系统很难吗?

结果通过pip安装pytorch的时候竟然报:

ERROR: Could not find a version that satisfies the requirement torch===1.6.0 (from versions: 0.1.2, 0.1.2.post1, 0.1.2.p
ost2)
ERROR: No matching distribution found for torch===1.6.0

这个…… 太让人无语了吧。

看到这里有网友进行了解释:https://stackoverflow.com/questions/58901682/could-not-find-a-version-that-satisfies-the-requirement-torch-1-3-1

所以老老实实地换回Python 3.7,并且注意选择了64位。

然后再安装,终于装好了。并且按照网友的介绍,配置了VS Code:https://zhuanlan.zhihu.com/p/31417084

没想到这样做也能踩到坑,估计还会有朋友和我遇到相同的问题,所以这篇博客应该还是有总结意义的。

整个过程主要参考了这两位网友的文章:https://www.pianshen.com/article/9805306909/ 以及 https://www.jianshu.com/p/d7851ffea38b。另外,如果大家想确定Tensor是否在GPU上,可以按照这个文章里的方法做一个简单判断:https://blog.csdn.net/qq_35435964/article/details/100928501

就简单记录这么多。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值