pytorch setup

来源：互联网发布：ubuntu 启动器图标编辑：程序博客网时间：2024/06/16 10:54

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

**
cuda7.5
conda install pytorch torchvision -c soumith

cuda8.0
conda install pytorch torchvision cuda80 -c soumith

阅读全文

0 0