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工具下载后离线安装.