最近深度学习需要用GPU版本的pytorch来加速运算,在csdn找了很多经验帖并试错后,终于明白了不出错的教程,用博客总结记录一下,以后帮别人安装可能还用的上,给需要的同学参考。
目标:
在Anaconda安装pytorch,能够在jupyter运行GPU版本的pytorch
前提:
已经正确安装好Anaconda,需要安装gpu版本的torch
最重要的是,电脑一定要有独立显卡。在帮室友安装的时候,发现显示适配器下面没有独立显卡,因为我们的电脑都是联想小新pro13,一开始是以为之前卸载了什么东西,所以一直在安装NVIDIA,后来安装的时候报错,版本和windows不匹配,认为是版本的问题。但是找了很久原因才发现,她的电脑是AMD芯片,并且不带有独立显卡,我的是Intel,型号是GeForce MX350。只能说对电脑的构造太不熟悉了。
安装:
下面的博客有检查显卡设备的方法,如果没有独立显卡,就可以暂停退出了。
https://blog.csdn.net/weixin_42838061/article/details/113107234
先按照步骤二、三、四安装,
注意:这一步选择network可能会报错,可以直接选local。具体原因我也不知道。
到了安装pytorch的步骤,就到下面这个链接:
packages
直接在官网下载包,找到对应的cuda版本和python版本,操作系统版本
我下载的如下:
cmd安装,找到所存放的文件夹,输入命令
pip install torch-1.7.1+cu101-cp39-cp39-win_amd64.whl
pip install torchvision-0.8.2+cu101-cp39-cp39-win_amd64.whl
pip install torchaudio-0.7.2-cp39-none-win_amd64.whl
torch就已经在Anaconda的site-packages了,再把包复制到虚拟环境也可以使用。验证方法如下:
已经帮很多个同学安装成功了,非常好用。
如有问题,欢迎评论区提出!