1、PyTorch 加速安装

---

一、通过 pip 安装

• 可提前在 https://download.pytorch.org/whl/torch_stable.html 下载好 pytorch 及 torchvision 相应版本的 whl 离线包
  
• 安装

# 1、安装 pip3
sudo apt install python3-pip

# 2、指定版本进行在线安装
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html

# 3、离线安装
pip install torch-1.7.1+cu101-cp38-cp38-linux_x86_64.whl  
pip install torchvision-0.8.2+cu101-cp38-cp38-linux_x86_64.whl

---

二、通过 conda 安装（清华源）

# 1、miniconda 下载与安装
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sh Miniconda3-latest-Linux-x86_64.sh

# 2、创建虚拟环境
conda create -n pytorch_1.7 python=3.7
conda activate pytorch_1.7

# 3、通过 -c 指定为 conda 自带的 pytorch 源（如果速度较慢，可切换清华源）
# conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch


# 4、验证安装是否成功（打印版本信息及 GPU 是否可用）
import torch
import torchvision

torch.__version__ 
>>> '1.7.1'

torchvision.__version__ 
>>> '0.8.2'

torch.cuda.is_available()
>>>True

---

三、通过官网 docker 镜像安装

• pytorch 官方镜像地址为：https://hub.docker.com/r/pytorch/pytorch/tags
• 选择 devel 版本，拉取镜像： docker pull pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
  

---

四、参考资料

1、https://pytorch.org/get-started/locally/
2、https://pytorch.org/get-started/previous-versions/
3、https://hub.docker.com/r/pytorch/pytorch/tags