ubuntu配置cudnn

Posted 2020-08-28 把全盛的爱都活过

参考网站：

http://blog.sina.com.cn/s/blog_a5fdbf010102w7f6.html

http://www.linuxidc.com/Linux/2015-04/116445.htm

ubuntu配置cudnn

1. 下载

https://developer.nvidia.com/rdp/cudnn-download

注册，下载，选择合适的版本。

用同样的方法尝试了cudnn-v3不行，cudnn-v4成功了。

1. 解压

将下载后的文件cudnn-7.0-linux-x64-v4.0-prod.solitairetheme8后缀名改为

cudnn-7.0-linux-x64-v4.0-prod.tgz

1. tar –xvf cudnn-7.0-linux-x64-v4.0-prod.tgz

1. 复制
   1. $cd cuda/include
   2. $sudo cp *.h /usr/local/cuda/include/
   3. $cd ../lib64
   4. $sudo cp lib* /usr/local/cuda/lib64/
2. 重新链接
   1. $cd /usr/local/cuda/lib64/
   2. $sudo rm -rf libcudnn.so libcudnn.so.4
   3. $sudo chmod u=rwx,g=rx,o=rx libcudnn.so.4.0.7
   4. $sudo ln -s libcudnn.so.4.0.7 libcudnn.so.4
   5. $sudo ln -s libcudnn.so.4 libcudnn.so
   6. $sudo ldconfig
3. 重新编译
   1. Cd caffe/
   2. Make clean
   3. Make all –j6

成功

1. 测试MNIST
   1. ./example/mnist/train_lenet.sh

发现仍然找不到libcudnn.so.4

Lib在编译时可以放在/usr/local/cuda/lib64

在系统运行时放在/usr/local/lib

因此，按照上述方法，重新将lib* copy到该目录。

再次运行，成功。48秒。