这个报错要么是环境变量配置问题,要么是CUDNN连接建立问题。
1.环境变量
在~/.bashrc 的最后添加
exportPATH=/usr/local/cuda/bin${PATH:+:${PATH}}
exportLD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
exportCUDA_HOME=/usr/local/cuda
2.CUDNN连接建立
cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有版本号,版本号在cudnn/lib64中查询
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软连接,注意自己下载的版本号
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig #立即生效
参考博客:http://www.bubuko.com/infodetail-2325869.html
版权声明:本文为weixin_40298200原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。