Utility

Linux安装cuDNN

  • 假设CUDA已经成功安装，先确定CUDA的版本

    cat /usr/local/cuda/version.txt

  • 进入cudnn官网，用户登录，在cudnn下载处根据Linux机器上的CUDA版本选择对应的cudnn版本

    1. 去nvidia cuDNN home page

    2. 点击下载

    3. 用户登录，完成问卷，提交

    4. 接受条款

    5. 选择与机器CUDA版本对应的cuDNN版本下载tar文件和Sample Code：

      cuDNN Library for Linux
      cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)

  • 使用tar文件安装

    1. 转到tar文件所在目录

    2. 解压

      $ tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

    3. 复制文件到已安装的CUDA Toolkit目录，并且修改权限

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

  • 为了测试安装成功否，安装samples code和用户指南：

    $ sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

  • 测试

    1. 把Samples Code 复制到可写的目录下$HOME

    2. 进入这个可写目录

      $ cd $HOME/cudnn_samples_v7/mnistCUDNN

    3. 编译 mnistCUDNN 实例

      $ make clean && make

    4. 运行可执行文件

      $ ./mnistCUDNN

      如果可以返回如下，表示安装成功：

      Test passed!