基于anaconda
基于python2.7,3.5,3.6,3.7 cuda10.1下 conda 安装命令(ubuntu和windows一样命令)
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
基于python2.7,3.5,3.6,3.7 cuda9.2下 conda 安装命令(ubuntu和windows一样命令)
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
pip安装方式 python2.7 cuda9.2
pip install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html
python3.5,3.6,3.7 cuda9.2
pip3 install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html
基于python2.7,3.5,3.6,3.7 cuda10.0下 conda 安装命令(ubuntu和windows一样命令)
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
python3.6 cuda10 仅linux
pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
python3.7 cuda10 仅linux
pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl
python3.7 cuda10.1 仅linux
pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
python3.7 cuda10.2 仅linux
pip install torch torchvision
python3.7 cuda11.0 仅linux
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
python3.7 cuda10 windows
pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-win_amd64.whl pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
基于python2.7,3.5,3.6,3.7 cuda9.0下 conda 安装命令(ubuntu和windows一样命令)
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
python3.7 cuda9 windows
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
基于python2.7,3.5,3.6,3.7 cuda9.1下 conda 安装命令(ubuntu和windows一样命令)
conda install pytorch cuda91 -c pytorch
Python3.6+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1,请选择对应的版本下载安装,不要同时执行下面三个命令!
pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
Python3.5+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1,请选择对应的版本下载安装,不要同时执行下面三个命令!
pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl
Python3.6+pip安装cpu版本
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
Python3.5+pip安装cpu版本
pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl
conda 安装 cpu版本 所有python版本通用 (windows和linux)
conda install pytorch-cpu -c pytorch