超详细的Pytorch GPU版本安装

最近深度学习需要用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了,再把包复制到虚拟环境也可以使用。验证方法如下:
在这里插入图片描述
已经帮很多个同学安装成功了,非常好用。
如有问题,欢迎评论区提出!

  • 6
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值