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環(huán)境) | 1、conda install pytorch-cpu -c pytorch 2羡洁、pip3 install torchvision |
- pip安裝法
2票顾、Linux安裝Pytorch
- conda安裝法
Python版本 | CUDA版本 | 指令 |
---|---|---|
Python2.7/3.5/3.6/3.7 | CUDA8.0 | conda install pytorch torchvision cuda80 -c pytorch |
Python3.5/3.6/3.7 | CUDA9.0 | conda install pytorch torchvision -c pytorch |
Python3.5/3.6/3.7 | CUDA9.2 | conda install pytorch torchvision cuda92 -c pytorch |
Python3.5/3.6/3.7 | None(即CPU環(huán)境) | conda install pytorch-cpu torchvision-cpu -c pytorch |
- pip安裝法
3、Mac安裝Pytorch
Mac OS二進(jìn)制文件不支持CUDA淤刃,如果需要CUDA晒他,則從源安裝
- conda安裝法
Python版本 | CUDA版本 | 指令 |
---|---|---|
Python2.7/3.5/3.6/3.7 | CUDA8.0/9.0/9.2/None | conda install pytorch torchvision -c pytorch |
- pip安裝法
Python版本 | CUDA版本 | 指令 |
---|---|---|
Python2.7 | CUDA8.0/9.0/9.2/None | pip install torch torchvision |
Python3.5/3.6/3.7 | CUDA8.0/9.0/9.2/None | pip3 install torch torchvision |