ubuntu系统cudnn安装及版本切换

一、引言

        在使用Tensorflow的过程中，由于版本问题，我们需要进行CUDNN版本的切换，在此大致描述安装匹配版本的cudnn，以及切换覆盖之前的cudnn的主要过程。

二、下载cudnn，解压拷贝

        百度cudnn，进入cudnn官网，找到我们需要适用的cudnn版本，在此举例，我要安装的是适配CUDA-8.0的cudnn6.0版本，官网注册账号（这里密码要求大小写及符号，比较复杂。。），下载对应版本cudnn。

       

        这里选择的是linux版本的cudnn，下载之后是一个后缀为tar的压缩包，从ubuntu的下载文档文件夹拷贝至home下，然后右键选择提取至此处，解压至当前文件夹，文件夹名为cuda。

        接下来需要做的是通过几步指令将需要的文件拷贝至系统的CUDA-8.0文件夹下。

$ sudo cp /home/wjx/cuda/include/cudnn.h /usr/local/cuda-8.0/include/

$ sudo cp /home/wjx/cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/

$ sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

三、建立cudnn软连接，切换至cudnn6.0

        由于之前系统cuda软连接已经和cudnn5.0匹配，所以需要先删除原有动态文件：

$ sudo rm -rf libcudnn.so libcudnn.so.5

        然后重新建立已有的cudnn6.0的链接库：

        这里6.0.21是我下载的库的具体名称，可能会有不同。

$ sudo ln -sf /usr/local/cuda-8.0/lib64/libcudnn.so.6.0.21 /usr/local/cuda-8.0/lib64/libcudnn.so.6

$ sudo ln -sf libcudnn.so.6 libcudnn.so

           软连接构建完毕，更新链接库：

$ sudo ldconfig

           到此为止，没有任何异常报错的话，新的cudnn6.0就配置好啦！

 

&另外，如果需要切换其他版本的cudnn时，基本步骤是一样的，只需重新下载拷贝文件，以及更新软连接库就行了。