第三步:cuDNN配置
cuDNN是GPU加速计算深层神经网络的库。
首先去官网(https://developer.nvidia.com/rdp/cudnn-archive)下载cuDNN,可能需要注册一个账号才能下载。根据自己的CUDA搭配相应的cuDNN库,如下图是我的搭配:
(1)下载基于Linux的库文件,我的电脑下载此文件的默认地址是/tmp/mozilla_y70000/ (根据自己的电脑确定)
(2)拷贝到需要目录,解压并修改权限操作,
chmod 777 cudnn-8.0-linux-x64-v5.1.tgz
tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
(3)进入cuDNN5.1的include目录,在命令行依次进行如下操作:
cd cuda/include
sudo cp cudnn.h /usr/local/cuda-8.0/include
cd ../lib64
sudo cp lib* /usr/local/cuda-8.0/lib64/
cd /usr/local/cuda-8.0/lib64/
sudo chmod +r libcudnn.so.5.1.10
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig
总计九个命令,执行完毕,即安装成功。
下一篇:【小卒CUDA第六篇】ubuntu16.04+CUDA8.0+GTX1050Ti+caffe的超详细配置(联想y7000+i7-8750H)之四
如果有更多ubuntu+cuda+caffe+slam方面的问题或者更多后续内容以及使用技巧请参考:https://blog.csdn.net/weixin_41994751