快速安装Pytorch和Torchvision

1 Linux下激活自己的虚拟环境并查看Python版本

如何创建虚拟环境见virtualenv创建虚拟环境

source myenv/bin/activate

激活环境后查看自己的python版本

python -V              # 我的是python3.6

2 查看需要安装的Pytorch和Torchvision版本

根据CUDA版本确定需要安装的Pytorch和Torchvision版本

https://pytorch.org/get-started/previous-versions/      # 官网，可能需要科学上网

3 直接命令行安装

3.1 如果不报错的话

以安装torch1.6.0为例

# -i 后面表示pip镜像源，不用镜像源太慢，本地以清华源为例，可更换
pip install torch==1.6.0 torchvision==0.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
# 若不知道源上有哪些版本，可采用如下命令查看返回情况
pip install torch== -i https://pypi.tuna.tsinghua.edu.cn/simple/

3.2 ERROR: Could not install packages due to an EnvironmentError: [Errno 28] No space left on device

错误原因：

• pip安装时需要将安装包首先解压到某个位置，默认目录是 /tmp，当默认目录无空间时就会出现该问题。
• 缓存空间不足

解决方案：

TMPDIR=/data/ pip install --cache-dir=/data/ --build /data/ torch==1.10.0

4 使用安装包本地安装

4.1本地安装原因

• 命令行安装网速原因报错
• pip源上找不到期待版本安装包，换源也行不通

4.2 从哪下载安装包

打开如下网址，其中有torch和torchvision安装包

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

根据自己的CUDA版本、Python版本、操作系统选择对应的安装包进行下载
图中cu101表示cuda10.1，cp36表示python3.6，linux表示linux操作系统

4.3 安装

在激活的虚拟环境中进行安装pytorch和torchvision

pip install torch-1.6.0-cp36-cp36m-linux_x86_64.whl
pip install  torchvision-0.7.0-cp36-cp36m-linux_x86_64.whl    # 自己的文件名即可

命令行中成功安装

4.4 测试是否安装成功

在python中运行如下代码

import torch
import torchvision
print(torch.__version__)
print(torch.cuda.is_available())