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

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/

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

猜你喜欢

转载自www.cnblogs.com/geoffreyone/p/10287879.html