【pytorch离线安装】【非常好用】pytorch离线安装

前(fei)言(hua):
  由于课题组服务器没联网(那点网速可以忽略不计),只好离线安装pytorch,毕竟平时也经常需要离线安装,真的是费了好大的劲。
  之前安装过,都失败了,最终如何解决的呢?哈哈~自己又网购了二手显卡,装到主机连上网,舒舒服服的使用官网安装命令安装。
   可是需要会的,永远躲不过去,不得已重新找方法。

一、正(gan)文(huo)

安装设备:ubuntu20.04+anaconda3
安装版本:pytorch1.6,其他版本类似

1、到pytorch官网:

  https://pytorch.org/get-started/locally/

在这里插入图片描述
此时展现的是最新的版本,我常用1.7的,因此点 install previous versions of PyTorch
在这里插入图片描述

2、找到linux下pip安装的命令:
# CUDA 11.0
pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 10.2
pip install torch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0

# CUDA 10.1
pip install torch==1.7.0+cu101 torchvision==0.8.0+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.7.0+cu92 torchvision==0.8.0+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.7.0+cpu torchvision==0.8.0+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

此时打开终端,输入nvidia-smi命令,查看你的驱动版本,以及cuda版本。此处的pytorch包中包含了cuda的相关东西,因此不需要再安装各种CUDA还是cudnn等的。

3、下载包:

我的是CUDA11.0,因此将对应的网址复制到浏览器,打开:

https://download.pytorch.org/whl/torch_stable.html

可以通过搜索获取需要的包torch以及torchvision(版本对应关系可以查看上述的pip命令中的版本对应),点击后下载信息会出现在浏览器下方,复制其下载链接,放在迅雷中下,速度嗖嗖的。
在这里插入图片描述
4、将包放在服务器的文件夹下,哪一个文件夹都可以。
使用source activate base激活anaconda默认环境base,如果显示没有此命令或者其他的,可以到anaconda3/bin下执行此命令。
之后执行:

pip install file:/data1/zhx/torch-1.7.1+cu110-cp37-cp37m-linux_x86_64.whl

pip install file:/data1/zhx/torchvision-0.8.2+cu110-cp37-cp37m-linux_x86_64.whl

/data1/zhx/是文件所在的路径。
这样就安装好了。

参考链接:
Python库的离线安装问题

猜你喜欢

转载自blog.csdn.net/YJYS_ZHX/article/details/115343707
今日推荐