linux pytorch 安装

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/AMDS123/article/details/69396953
http://pytorch.org/

conda

cuda7.5 
conda install pytorch torchvision -c soumith

cuda8.0 
conda install pytorch torchvision cuda80 -c soumith


pip

cuda7.5

python2.7
pip install http://download.pytorch.org/whl/cu75/torch-0.1.11.post5-cp27-none-linux_x86_64.whl 
pip install torchvision 

python3.5
pip install http://download.pytorch.org/whl/cu75/torch-0.1.11.post5-cp35-cp35m-linux_x86_64.whl 
pip install torchvision

python3.6
pip install http://download.pytorch.org/whl/cu75/torch-0.1.11.post5-cp36-cp36m-linux_x86_64.whl
pip install torchvision



cuda8.0

python2.7

pip install http://download.pytorch.org/whl/cu80/torch-0.1.11.post5-cp27-none-linux_x86_64.whl
pip install torchvision

python3.5
pip install http://download.pytorch.org/whl/cu80/torch-0.1.11.post5-cp35-cp35m-linux_x86_64.whl
pip install torchvision

python3.6
pip install http://download.pytorch.org/whl/cu80/torch-0.1.11.post5-cp36-cp36m-linux_x86_64.whl
pip install torchvision

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页