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

1:登录Pytorch官网

官网地址:https://pytorch.org/
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第1张图片

2:选择Pytorch版本信息

点击install,进入安装页面
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第2张图片
这里我选择的是Pytorch 1.9.0版,CUDA版本号为11.1

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

这里我的显卡是n卡GTX 1650
win+R打开cmd命令行,输入命令nvidia-smi
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第3张图片
确保显卡驱动版本大于400才可,我的CUDA版本为11.1

(2)若显卡不支持CUDA,则选择cpu版本即可
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第4张图片

3:下载并安装Pytorch包

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

于是我尝试用离线安装包在本地安装:
(1)在Pytorch官网安装页面上改为pip安装
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第6张图片
注意图中红线位置的版本号,我的为:torch==1.9.0+cu111 torchvision==0.10.0+cu111
然后去离线下载网址下载对应的torch和torchvision安装包,推荐用迅雷下载速度快。
下载完成后如图:
在这里插入图片描述

(2)打开命令行,创建虚拟环境
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第7张图片
(3)进入环境
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第8张图片
(4)输入命令:pip install (whl文件保存路径)\文件名称
分别安装两个文件
Pytorch深度学习框架安装(图文详解,本地安装包不使用官网和镜像源)(更新于2021.9.18)_第9张图片
注意:还是需要联网安装,因为安装包可能会有更新,会下载一部分更新后的包!!!

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

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

你可能感兴趣的:(Python,pytorch,深度学习,python)