首页 > 代码库 > ubuntu配置cudnn
ubuntu配置cudnn
参考网站:
http://blog.sina.com.cn/s/blog_a5fdbf010102w7f6.html
http://www.linuxidc.com/Linux/2015-04/116445.htm
ubuntu配置cudnn
- 下载
https://developer.nvidia.com/rdp/cudnn-download
注册,下载,选择合适的版本。
用同样的方法尝试了cudnn-v3不行,cudnn-v4成功了。
- 解压
将下载后的文件cudnn-7.0-linux-x64-v4.0-prod.solitairetheme8后缀名改为
cudnn-7.0-linux-x64-v4.0-prod.tgz
- tar –xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
- 复制
- $cd cuda/include
- $sudo cp *.h /usr/local/cuda/include/
- $cd ../lib64
- $sudo cp lib* /usr/local/cuda/lib64/
- 重新链接
- $cd /usr/local/cuda/lib64/
- $sudo rm -rf libcudnn.so libcudnn.so.4
- $sudo chmod u=rwx,g=rx,o=rx libcudnn.so.4.0.7
- $sudo ln -s libcudnn.so.4.0.7 libcudnn.so.4
- $sudo ln -s libcudnn.so.4 libcudnn.so
- $sudo ldconfig
- 重新编译
- Cd caffe/
- Make clean
- Make all –j6
成功
- 测试MNIST
- ./example/mnist/train_lenet.sh
发现仍然找不到libcudnn.so.4
Lib在编译时可以放在/usr/local/cuda/lib64
在系统运行时放在/usr/local/lib
因此,按照上述方法,重新将lib* copy到该目录。
再次运行,成功。48秒。
ubuntu配置cudnn
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。