Ubuntu16.04下离线安装pytorch

问题背景

最近在安装pytorch的时候总遇到网络问题,在线安装动不动就失败,即使更换了清华源也会出现一样的问题,于是打算先下载pytorch的安装文件,再进行离线安装,简单记录一下。
参考博客:https://www.cnblogs.com/darkknightzh/p/12000809.html

下载pytorch和torchvision

去清华的镜像网站下载pytorch和torchvision,根据你需要的环境下载不同的版本,我下载的是pytorch-1.6.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2torchvision-0.7.0-py37_cu101.tar.bz2
清华大学镜像

下载cudatoolkit

去conda官网下载cudatoolkit,我下载的是cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
cudatoolkit下载地址

安装pytorch、torchvision、cudatoolkit

 conda install --offline pytorch-1.6.0-py3.7_cuda10.1.243_cudnn7.6.3_0.tar.bz2 
 conda install --offline torchvision-0.7.0-py37_cu101.tar.bz2 
 conda install --offline cudatoolkit-10.1.243-h6bb024c_0.tar.bz2 

安装完毕以后运行安装命令来匹配之前离线安装好的东西

conda install pytorch torchvision cudatoolkit=10.1

检验是否安装成功

在这里插入图片描述
出现以上结果即为安装成功,torch.cuda.is_available() 代表可以使用GPU,如果是false,意味着只能使用CPU。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值