问题描述:
Linux下,使用pip 离线安装pytorch并不会自动安装torchvision。
ps. 文章最后给出在线安装步骤
完整安装步骤
-
进入官网:点击红字,选择安装之前的版本。
-
定位到V1.1.0,第一大项为conda安装,我们使用Wheel(即pip安装)。Linux and Windows下发现下载网址https://download.pytorch.org/whl/cu100/torch_stable.html(要安装cuda 9.0,把cu100改为cu90就好),复制到浏览器打开。
以下可直接复制:
# 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
- 下载Linux版pytorch==1.1.0
cp36对应python3.6(本文已安装python3.6)
- 在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版本
发现torchvision>=0.4.0版本都会自动安装对应的pytorch版本。
因此这里安装torchvision==0.3.0
安装成功。
在线安装
直接输入命令:
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未安装,请查看上述步骤进行安装。