Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)

1:登录Pytorch官网

官网地址:https://pytorch.org/
在这里插入图片描述

2:选择Pytorch版本信息

点击install,进入安装页面
在这里插入图片描述
这里我选择的是Pytorch 1.9.0版,CUDA版本号为11.1

如何查看CUDA的版本信息呢?
(1)首先确定你的显卡支持CUDA,这个可以百度

这里我的显卡是n卡GTX 1650
win+R打开cmd命令行,输入命令nvidia-smi
在这里插入图片描述
确保显卡驱动版本大于400才可,我的CUDA版本为11.1

(2)若显卡不支持CUDA,则选择cpu版本即可
在这里插入图片描述

3:下载并安装Pytorch包

复制安装页面上的命令
在这里插入图片描述
打开命令行进入Python的虚拟环境,粘贴上面的命令,于是开始了最烦的一步,因为这个官网服务器在海外,国内大陆下载速度很慢不说,还经常下到一般断开,因为命令行中下载不支持断点续连,所以一旦下载失败,我连Anaconda都要重新装一遍,于是开始了漫长的下载-重装-下载-重装,耗费我一下午时间还没有装好,用清华、中科大等镜像也无法解决问题。

于是我尝试用离线安装包在本地安装:
(1)在Pytorch官网安装页面上改为pip安装
在这里插入图片描述
注意图中红线位置的版本号,我的为:torch==1.9.0+cu111 torchvision==0.10.0+cu111
然后去离线下载网址下载对应的torch和torchvision安装包,推荐用迅雷下载速度快。
下载完成后如图:
在这里插入图片描述

(2)打开命令行,创建虚拟环境
在这里插入图片描述
(3)进入环境
在这里插入图片描述
(4)输入命令:pip install (whl文件保存路径)\文件名称
分别安装两个文件
在这里插入图片描述
注意:还是需要联网安装,因为安装包可能会有更新,会下载一部分更新后的包!!!

(5)检验安装是否成功
在这里插入图片描述

进入python 控制台
import torch
无报错
torch.cuda.is_available()
返回True
安装成功!!!

  • 4
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

是小猪猪哦

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值