1、windows安装Pytorch
- conda安装法
Python版本 | CUDA版本 | 指令 |
---|---|---|
Python3.5/3.6/3.7 | CUDA8.0 | 1、conda install pytorch cuda80 -c pytorch 2、pip3 install torchvision |
Python3.5/3.6/3.7 | CUDA9.0 | 1、conda install pytorch -c pytorch 2、pip3 install torchvision |
Python3.5/3.6/3.7 | CUDA9.2 | 1、conda install pytorch cuda92 -c pytorch 2、pip3 install torchvision |
Python3.5/3.6/3.7 | None(即CPU环境) | 1、conda install pytorch-cpu -c pytorch 2、pip3 install torchvision |
- pip安装法
Python版本 | CUDA版本 | 指令 |
---|---|---|
Python3.5 | CUDA8.0 | 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl 2、pip3 install torchvision |
Python3.6 | CUDA8.0 | 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl 2、pip3 install torchvision |
Python3.7 | CUDA8.0 | 1、pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl< |