Ubuntu16.04上安装cudnn教程和opencv

https://blog.csdn.net/wang15061955806/article/details/80791112

Ubuntu16.04上安装cudnn教程

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

查看了caffe最新版程序的install requirement : https://github.com/BVLC/caffe,这里我安装的是CUDA8.0+cudnn5.0(注:其实使用cudnn6.0也可以,但是后来安装tensorfow-gpu1.0.1时,出现cudnn版本问题,其只支持cudnn5.0,似乎要到tensorflow1.3.0后才支持cudnn6.0,反正现在cudnn5.0也够用了!)

下载完成后解压,得到一个 cudn 文件夹,该文件夹下include 和 lib64 两个文件夹,命令行进入 cudn/include 路径下,然后进行以下操作:

sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
  • 1

然后命令行进入 cudn/lib64 路径下,运行以下命令:

  1. sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
  2. cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
  3. sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成软衔接
  4. sudo ln -s libcudnn.so.6 libcudnn.so #生成软链接

安装完成后可用 nvcc -V 命令验证是否安装成功,若出现以下信息则表示安装成功:

  1. yhao @yhao-X550VB:~$ nvcc -V
  2. nvcc: NVIDIA (R) Cuda compiler driver
  3. Copyright (c) 2005-2016 NVIDIA Corporation
  4. Built on Tue_Jan_10_13:22:03_CST_2017
  5. Cuda compilation tools, release 8.0, V8.0.61
     
     
    opencv安装部分参考博客:
    https://blog.csdn.net/cocoaqin/article/details/78163171

猜你喜欢

转载自www.cnblogs.com/shuimuqingyang/p/10009600.html