1. 程式人生 > >pip安裝Python3.6下各版本pytorch命令

pip安裝Python3.6下各版本pytorch命令

Python3.6 + cuda none
pip3 install http://download.pytorch.org/whl/cpu/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
 
pip3 install torchvision




Python3.6 + cuda 8
pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
 
pip3 install torchvision




Python3.6 + cuda 9
pip3 install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
 
pip3 install torchvision


Python3.6 + cuda 9.1
pip3 install http://download.pytorch.org/whl/cu91/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
 
pip3 install torchvision