pytorch 安装 及对应版本

可以自行去官网查看 https://pytorch.org/get-started/previous-versions/
v1.8.0
Conda
OSX

conda

conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 -c pytorch
Linux and Windows

CUDA 10.2

conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

CUDA 11.1

conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

CPU Only

conda install pytorch1.8.0 torchvision0.9.0 torchaudio0.8.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.8.0 torchvision0.9.0 torchaudio0.8.0
Linux and Windows

RocM 4.0.1 (Linux only)

pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
pip install ninja
pip install ‘git+https://github.com/pytorch/[email protected]’

CUDA 11.1

pip install torch1.8.0+cu111 torchvision0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 10.2

pip install torch1.8.0 torchvision0.9.0 torchaudio==0.8.0

CPU only

pip install torch1.8.0+cpu torchvision0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
v1.7.1
Conda
OSX

conda

conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

CUDA 10.2

conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch

CUDA 11.0

conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

CPU Only

conda install pytorch1.7.1 torchvision0.8.2 torchaudio0.7.2 cpuonly -c pytorch
Wheel
OSX
pip install torch1.7.1 torchvision0.8.2 torchaudio0.7.2
Linux and Windows

CUDA 11.0

pip install torch1.7.1+cu110 torchvision0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 10.2

pip install torch1.7.1 torchvision0.8.2 torchaudio==0.7.2

CUDA 10.1

pip install torch1.7.1+cu101 torchvision0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 9.2

pip install torch1.7.1+cu92 torchvision0.8.2+cu92 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.7.1+cpu torchvision0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
v1.7.0
Conda
OSX

conda

conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

CUDA 10.2

conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch

CUDA 11.0

conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

CPU Only

conda install pytorch1.7.0 torchvision0.8.0 torchaudio0.7.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.7.0 torchvision0.8.0 torchaudio0.7.0
Linux and Windows

CUDA 11.0

pip install torch1.7.0+cu110 torchvision0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 10.2

pip install torch1.7.0 torchvision0.8.0 torchaudio==0.7.0

CUDA 10.1

pip install torch1.7.0+cu101 torchvision0.8.0+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 9.2

pip install torch1.7.0+cu92 torchvision0.8.0+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.7.0+cpu torchvision0.8.0+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
v1.6.0
Conda
OSX

conda

conda install pytorch1.6.0 torchvision0.7.0 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.6.0 torchvision0.7.0 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.6.0 torchvision0.7.0 cudatoolkit=10.1 -c pytorch

CUDA 10.2

conda install pytorch1.6.0 torchvision0.7.0 cudatoolkit=10.2 -c pytorch

CPU Only

conda install pytorch1.6.0 torchvision0.7.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.6.0 torchvision0.7.0
Linux and Windows

CUDA 10.2

pip install torch1.6.0 torchvision0.7.0

CUDA 10.1

pip install torch1.6.0+cu101 torchvision0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 9.2

pip install torch1.6.0+cu92 torchvision0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.6.0+cpu torchvision0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
v1.5.1
Conda
OSX

conda

conda install pytorch1.5.1 torchvision0.6.1 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.5.1 torchvision0.6.1 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.5.1 torchvision0.6.1 cudatoolkit=10.1 -c pytorch

CUDA 10.2

conda install pytorch1.5.1 torchvision0.6.1 cudatoolkit=10.2 -c pytorch

CPU Only

conda install pytorch1.5.1 torchvision0.6.1 cpuonly -c pytorch
Wheel
OSX
pip install torch1.5.1 torchvision0.6.1
Linux and Windows

CUDA 10.2

pip install torch1.5.1 torchvision0.6.1

CUDA 10.1

pip install torch1.5.1+cu101 torchvision0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 9.2

pip install torch1.5.1+cu92 torchvision0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.5.1+cpu torchvision0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
v1.5.0
Conda
OSX

conda

conda install pytorch1.5.0 torchvision0.6.0 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.5.0 torchvision0.6.0 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.5.0 torchvision0.6.0 cudatoolkit=10.1 -c pytorch

CUDA 10.2

conda install pytorch1.5.0 torchvision0.6.0 cudatoolkit=10.2 -c pytorch

CPU Only

conda install pytorch1.5.0 torchvision0.6.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.5.0 torchvision0.6.0
Linux and Windows

CUDA 10.2

pip install torch1.5.0 torchvision0.6.0

CUDA 10.1

pip install torch1.5.0+cu101 torchvision0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

CUDA 9.2

pip install torch1.5.0+cu92 torchvision0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.5.0+cpu torchvision0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
v1.4.0
Conda
OSX

conda

conda install pytorch1.4.0 torchvision0.5.0 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.4.0 torchvision0.5.0 cudatoolkit=9.2 -c pytorch

CUDA 10.1

conda install pytorch1.4.0 torchvision0.5.0 cudatoolkit=10.1 -c pytorch

CPU Only

conda install pytorch1.4.0 torchvision0.5.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.4.0 torchvision0.5.0
Linux and Windows

CUDA 10.1

pip install torch1.4.0 torchvision0.5.0

CUDA 9.2

pip install torch1.4.0+cu92 torchvision0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.4.0+cpu torchvision0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
v1.2.0
Conda
OSX

conda

conda install pytorch1.2.0 torchvision0.4.0 -c pytorch
Linux and Windows

CUDA 9.2

conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=9.2 -c pytorch

CUDA 10.0

conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=10.0 -c pytorch

CPU Only

conda install pytorch1.2.0 torchvision0.4.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.2.0 torchvision0.4.0
Linux and Windows

CUDA 10.0

pip install torch1.2.0 torchvision0.4.0

CUDA 9.2

pip install torch1.2.0+cu92 torchvision0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

CPU only

pip install torch1.2.0+cpu torchvision0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
v1.1.0
Conda
OSX

conda

conda install pytorch1.1.0 torchvision0.3.0 -c pytorch
Linux and Windows

CUDA 9.0

conda install pytorch1.1.0 torchvision0.3.0 cudatoolkit=9.0 -c pytorch

CUDA 10.0

conda install pytorch1.1.0 torchvision0.3.0 cudatoolkit=10.0 -c pytorch

CPU Only

conda install pytorch-cpu1.1.0 torchvision-cpu0.3.0 cpuonly -c pytorch
Wheel
OSX
pip install torch1.1.0 torchvision0.3.0
Linux and Windows

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
v1.0.1
Conda
OSX

conda

conda install pytorch1.0.1 torchvision0.2.2 -c pytorch
Linux and Windows

CUDA 9.0

conda install pytorch1.0.1 torchvision0.2.2 cudatoolkit=9.0 -c pytorch

CUDA 10.0

conda install pytorch1.0.1 torchvision0.2.2 cudatoolkit=10.0 -c pytorch

CPU Only

conda install pytorch-cpu1.0.1 torchvision-cpu0.2.2 cpuonly -c pytorch
Wheel
OSX
pip install torch1.0.1 torchvision0.2.2
Linux and Windows

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
v1.0.0
Conda
OSX

conda

conda install pytorch1.0.0 torchvision0.2.1 -c pytorch
Linux and Windows

CUDA 10.0

conda install pytorch1.0.0 torchvision0.2.1 cuda100 -c pytorch

CUDA 9.0

conda install pytorch1.0.0 torchvision0.2.1 cuda90 -c pytorch

CUDA 8.0

conda install pytorch1.0.0 torchvision0.2.1 cuda80 -c pytorch

CPU Only

conda install pytorch-cpu1.0.0 torchvision-cpu0.2.1 cpuonly -c pytorch
Wheel
OSX
pip install torch1.0.0 torchvision0.2.1
Linux and Windows

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

CUDA 8.0

Download and install wheel from https://download.pytorch.org/whl/cu80/torch_stable.html

CPU only

Download and install wheel from https://download.pytorch.org/whl/cpu/torch_stable.html