本文主要是介绍ubuntu pytorch setup,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
**
pytorch setup
conda安装失败了,最后那还是用pip成功!
**
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
**
conda
**
cuda7.5
conda install pytorch torchvision -c soumith
cuda8.0
conda install pytorch torchvision cuda80 -c soumith
这篇关于ubuntu pytorch setup的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!