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 直接命令行安装
以安装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/
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操作系统
Pytorch安装包选择
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())
