基本环境:Ubuntu14.04.5(64位)+ Python27 + CUDA8.0
Pytorch安装方式
<1> 首推:用conda安装(版本号需要自己给出)
<1.1> 已经安装过CUDA的,根据不同的CUDA版本选择下面的命令
conda install pytorch=0.1.12 cuda80 -c soumith
conda install pytorch=0.1.12 cuda75 -c soumith
pytorch=x.x.x 根据自己的需求,表明自己需要的版本号:目前的版本号有:0.1.12, 0.2.0, 0.3.0 等
<1.2> 目前还没有安装CUDA
conda install pytorch=x.x.x -c soumith 版本号同上,自定义
<2> 其次:用pip安装
<2.1> 下载whl格式安装包,离线安装: pip install /path/to/whl/file
<2.2> 给出whl安装包网络,在线安装:
pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl
各种版本的whl的网络路径(下载地址)在这里找到:点这里,需要提醒,有操作系统版本不同,不同Python版本,CUDA支持与否
manylinux1_x86_64(带有manylinux1标识)是不支持CUDA的
<3> 不推荐:源码编译安装,给个链接
主要参考了官方的安装文档
如果不小心被你看到了^_^,欢迎交流你遇到的问题(留言,私信都可以)