Ubuntu14.04+GTX980 Ti+cuDNN 5.1

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_17448289/article/details/56678555

升级gcc 4.9+

1. 查看gcc/g++版本

ubuntu14.04 自带的是4.8版本,而最新的4.9.2对c++11的支持更好一些。

gcc --version

这里写图片描述

2. 升级gcc 4.9+

cuDNN安装

1. cuDNN需要注册Accelerated Computing Developer Program,然后可以免费下载。

cuDNN是The NVIDIA CUDA Deep Neural Network library,对于使用NVIDIA显卡进行深度学习加速具有很大的性能提升,非常建议添加。

PS:另外,cuDNN在很多工程中兼容性较差,可能需要安装特定的历史版本

2. 进入cuDNN官网地址,点Download进入下载界面

这里写图片描述

根据自己情况选择,博主选择了cuDNN v5.1 for Linux,下载,大约77.2M, cudnn-7.5-linux-x64-v5.1.tgz

3. 解压cudnn-7.5-linux-x64-v5.1.tgz,文件夹名为“cuda”,得到cudnn的头文件cudnn.h (在~/include/ 中);以及库文件libcudnn.so libcudnn.so.5 libcudnn.so.5.1.10 libcudnn_static.a

其中:
1. libcudnn.so.5.1.10是动态链接库文件
2. libcudnn.so.5 是指向libcudnn.so.5.1.10的符号链接文件
3. libcudnn.so是指向libcudnn.so.5 的符号链接文件
4. libcudnn_static.a 是静态库文件

将头文件拷贝到cuda安装目录下的include,我的是 /usr/local/cuda/include/

将库文件拷贝到cuda安装目录下的lib64,我的是/usr/local/cuda/lib64/

先进入cuda的lib64文件夹下执行:
sudo cp -r lib* /usr/local/cuda/lib64/
cd include  
sudo cp cudnn.h /usr/local/cuda/include/

/usr/local/cuda/lib64/这个文件夹下还有其他目录,所以cp的时候要加上-r,递归复制进去。
/usr/local/cuda/lib64/里面复制过来的文件:
这里写图片描述

在/usr/local/cuda/include/下可以找到我们复制过来的cudnn.h:
这里写图片描述

4. 更新软链接

注意拷贝完之后,符号链接文件要重新建立,使用如下命令重新建立符号链接文件

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so

PS:根据你下载的cuDNN版本不同,需要对如上命令中的版本进行修改,以上展示的命令是对于cuDNN 5.1.10的。

猜你喜欢

转载自blog.csdn.net/qq_17448289/article/details/56678555
今日推荐