Ubuntu18.04配置python2.7+torch0.3.1虚拟环境
1.创建anaconda虚拟环境,并同时安装python2.7
conda create -n py2.7 python=2.7
2.下载torch0.3.1到本地
链接如下(此torch版本是cuda9.0编译的。由于有的显卡必须要cuda9.0以上才兼容所以相应的torch也得符合cuda的版本,所以特别说明这一点。我的环境如果直接pip install torch==0.3.1是默认安装的cuda8.0编译的与我的显卡不匹配,运行时会报错,因此需要安装cuda9.0编译的torch,如果你也有这样的情况可以试一试。):
3.在下载好的.whl文件目录下打开终端,进入创建好的名为py2.7的虚拟环境中
source activate py2.7
pip install torch-0.3.1-cp27-cp27mu-linux_x86_64.whl