Ubuntu16.04安装PyTorch

注: 首先确保已安装好nvidia驱动和cuda，通过如下命令查看是否已安装。

nvidia-smi

nvcc -V

1、安装Anaconda

清华源镜像下载地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

下载安装包进行安装

bash Anaconda3-5.3.1-Linux-x86_64.sh
# ENTER
# yes
# ENTER
# yes
# no

2、在线pip安装

# 从官网选择相应cuda版本的安装命令
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
# -i https://mirrors.aliyun.com/pypi/simple/ 阿里源
# -i https://pypi.tuna.tsinghua.edu.cn/simple 清华源

3、离线安装torch和torchvision

镜像下载地址: https://download.pytorch.org/whl/torch_stable.html
分别下载torch和torchvision


开始安装torch和torchvision

conda create -n pytorch python=3.7
conda activate pytorch
cd <torch directory> #cd进入torch和torchvision所在目录
pip install torch-1.6.0-cp37-cp37m-linux_x86_64.whl -i https://pypi.doubanio.com/simple
pip install torchvision-0.7.0-cp37-cp37m-linux_x86_64.whl -i https://pypi.doubanio.com/simple

注: 安装过程中如果出现pip需要升级，则可以如下操作:

pip install --upgrade pip -i https://pypi.doubanio.com/simple

4、验证及再次激活PyTorch

source activate pytorch
python3 #进入python环境
import torch as t
print(t.cuda.is_available()) #若输出为True则说明安装成功