PyTorch在Windows,Linux,Mac(conda/
作者:
初级小白程序媛 | 来源:发表于
2018-10-22 21:21 被阅读0次
1、windows安装Pytorch
| 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 |
2、Linux安装Pytorch
| 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环境) |
conda install pytorch-cpu torchvision-cpu -c pytorch |
3、Mac安装Pytorch
Mac OS二进制文件不支持CUDA，如果需要CUDA，则从源安装
| Python版本 |
CUDA版本 |
指令 |
| Python2.7/3.5/3.6/3.7 |
CUDA8.0/9.0/9.2/None |
conda install pytorch torchvision -c pytorch |
| 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 |
本文标题：PyTorch在Windows,Linux,Mac(conda/
本文链接：https://www.haomeiwen.com/subject/yyjnzftx.html
网友评论