您的位置:首页 > 其它

ubuntu安装pytorch以及torchnet

2017-11-14 09:32 2541 查看
http://pytorch.org/
转自http://blog.csdn.net/amds123/article/details/69396953
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

安装torchnet 参考https://github.com/pytorch/tnt
上面一系列装好之后,发现在python中还是没办法import torchnet
首先得Make sure you have PyTorch installed, then do:
pip install git+https://github.com/pytorch/tnt.git@master
应该ok了,至于torch,我一直很奇怪,和pytorch还是有点区别的
pip install --upgrade git+https://github.com/pytorch/tnt.git@master
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: