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