Pytorch安装(基于anaconda虚拟环境)

Pytorch安装倒腾了一上午终于搞定,记录一下安装过程.

1. 首先尝试官网的安装方式,但是网速太慢了. 除去cudnn100, torchvision和pytorch三个文件,其余可以直接从清华镜像源下载.

　　这三个包直接进入官网,使用多线程(我用idm下载的,最好在挂个代理或者VPN)下载后,离线安装.离线安装命令:

conda install --use-local Pkg_name

2. 安装完成后报错,拷贝系统lib目录下相关文件到需求路径,测试成功.

执行如下代码,需要根据安装路径适当调整:

cp /usr/lib/libstdc++.so.6 /opt/anaconda/envs/torch_36/lib/python3.6/site-packages/torch/lib/

cp /usr/lib/libgomp.so.1 /opt/anaconda/envs/torch_36/lib/python3.6/site-packages/torch/lib/

拷贝完毕后进行测试,如下:

 这样子测试GPU:

torch.cuda.is_available()

---

更新时间:2019-1-24 上面一个一个安装太费事了,还是新建一个虚拟环境,直接用

pip install -r requirements.txt

比较方便.其中torch下载太慢了,我拷贝链接,用axel工具下载后离线安装.