PyTorch 安装命令合集

Torch 1.0.0 版本将下文中torch-0.4.1更换为torch-1.0.0

Linux

使用 conda
以下命令适用于 Python 2.7 - 3.7。

CPU 版本：
conda install pytorch-cpu torchvision-cpu -c pytorch

CUDA 8：
conda install pytorch torchvision cuda80 -c pytorch

CUDA 9.0：
conda install pytorch torchvision -c pytorch

CUDA 9.2：
conda install pytorch torchvision cuda92 -c pytorch

使用 pip
Python 2.7

CPU 版本：
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注：if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

CUDA 8：
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注：if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

CUDA 9.0：
pip install torch torchvision

CUDA 9.2：
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision
注：if the above command does not work, then you have python 2.7 UCS2, use this command
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp27-cp27m-linux_x86_64.whl

Python 3.5
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install torch torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision

Python 3.6
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install torch torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

Python 3.7
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install torch torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1.post2-cp37-cp37m-linux_x86_64.whl
pip3 install torchvision

MacOS

使用 conda
以下命令适用于 Python 2.7 - 3.7。

CPU 版本：
conda install pytorch torchvision -c pytorch
CUDA 8，CUDA 9.0，或 CUDA 9.2：MacOS 不支持 CUDA，如果需要，则需要从源码编译安装。

使用 pip
Python 2.7
CPU 版本：
pip install torch torchvision
CUDA 8，CUDA 9.0，或 CUDA 9.2：MacOS 不支持 CUDA，如果需要，则需要从源码编译安装。

Python 3.5，3.6，3.7
CPU 版本：
pip3 install torch torchvision
CUDA 8，CUDA 9.0，或 CUDA 9.2：MacOS 不支持 CUDA，如果需要，则需要从源码编译安装。

Windows

使用 conda
Python 2.7
PyTorch 不支持在 Windows 上的 Python 2.7 版本，请使用 Python 3 版本。

Python 3.5，3.6，3.7
CPU 版本：
conda install pytorch-cpu -c pytorch
pip3 install torchvision

CUDA 8：
conda install pytorch cuda80 -c pytorch
pip3 install torchvision

CUDA 9.0：
conda install pytorch -c pytorch
pip3 install torchvision

CUDA 9.2：
conda install pytorch cuda92 -c pytorch
pip3 install torchvision

使用 pip
Python 2.7
PyTorch 不支持在 Windows 上的 Python 2.7 版本，请使用 Python 3 版本。

Python 3.5
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

Python 3.6
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

Python 3.7
CPU 版本：
pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 8：
pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 9.0：
pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

CUDA 9.2：
pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision