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

烂漫一生 提交于 2021-02-08 07:23:36

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

 

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

  这三个包直接进入官网,使用多线程(我用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工具下载后离线安装.

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!