Linux服务器安装pytorch1.1.0安装步骤及问题解决

问题描述:

Linux下,使用pip 离线安装pytorch并不会自动安装torchvision。
ps. 文章最后给出在线安装步骤

完整安装步骤

  1. 进入官网:点击红字,选择安装之前的版本
    Linux服务器安装pytorch1.1.0安装步骤及问题解决_第1张图片

  2. 定位到V1.1.0,第一大项为conda安装,我们使用Wheel(即pip安装)。Linux and Windows下发现下载网址https://download.pytorch.org/whl/cu100/torch_stable.html(要安装cuda 9.0,把cu100改为cu90就好),复制到浏览器打开。
    Linux服务器安装pytorch1.1.0安装步骤及问题解决_第2张图片
    以下可直接复制:

# CUDA 10.0
Download and install wheel from https://download.pytorch.org/whl/cu100/torch_stable.html

# CUDA 9.0
Download and install wheel from https://download.pytorch.org/whl/cu90/torch_stable.html

# CPU only
Download and install wheel from https://download.pytorch.org/whl/cpu/torch_stable.html
  1. 下载Linux版pytorch==1.1.0
    Linux服务器安装pytorch1.1.0安装步骤及问题解决_第3张图片

cp36对应python3.6(本文已安装python3.6)

  1. 在Linux安装(本文已使用Xshell连接到远程服务器)
    安装命令:
    pip install torch-1.1.0-cp36-cp36m-linux_x86_64.whl
    注:如果.whl文件是放到其他文件夹,请定位到该文件夹路径进行安装

特别注意

安装完成后,使用pip list,发现没有torchvision,因为使用pip离线安装不会自动安装torchvision。
并且使用pip instal torchvison会自动安装最新版的torchvison,且会安装最新版的pytorch(此处为1.14)

在这里插入图片描述
CTRL+C中止。
安装指定版本pip install torchvision==0.4.0
发现会自动安装pytorch==1.2.0版本

Linux服务器安装pytorch1.1.0安装步骤及问题解决_第4张图片

发现torchvision>=0.4.0版本都会自动安装对应的pytorch版本。
因此这里安装torchvision==0.3.0

Linux服务器安装pytorch1.1.0安装步骤及问题解决_第5张图片
安装成功。

在线安装

直接输入命令:
pip install -U https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl

如果安装cuda 9.0版本,把cu100改为cu90
安装其他pytorch版本,在1.1.0对应位置修改即可

安装过程中下载缓慢,请添加Pytorch的清华源(具体搜素其他博客)。完成后,pip list 下,看是否安装好了torch和torchvision,若torchvision未安装,请查看上述步骤进行安装。

你可能感兴趣的:(Linux,各种问题)