deeplearnig

cudnn Installation in Linux

cudnn Installation in Linux

安装教程相对简单,实际原理就是将 cudnn的库加入 cudaincludelib 文件夹中

0.前提

  1. 此方法适用于Linux 系统,常见的 Ubuntu ,Debian 等等
  2. 首先你的系统已经成功安装好 CUDA toolkit
  3. 从官方网站上下载 cudnn 的压缩包(很多版本,下载你需要的),需要登录账户

1. Ubuntu 下安装

  1. 进入存在 cudnn**.tgz 的文件夹

  2. 解压

    
    $ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
    
  3. 将解压过的文件,复制到你cuda的安装目录即可

    
    $ 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*
    

2. Debian 下安装

Debian下的安装方法跟简单,直接安装打包好的.deb就可以了.

  1. 进入debian版本cudnn的文件夹下

  2. 安装运行时的库

    
    sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
    
  3. 安装开发者的库

    
    sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
    
  4. 安装 例程和用户指南

    
    sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
    

3.测试一下是否安装成功

运行一个小Demo即可.

如果安装了 例程和用户指南 这个包的话,我们可以找到位于 /usr/src/cudnn_samples_v7mnistCUDNN这个小例子.

  1. 拷贝到 你的home/yourdir 任意文件夹下

    
    $cp -r /usr/src/cudnn_samples_v7/ $HOME
    
  2. 进入 mnistCUDNN

    
    $ cd $HOME/cudnn_samples_v7/mnistCUDNN
    
  3. 编译

    
    $make clean && make
    
  4. 运行

    
    $ ./mnistCUDNN
    
  5. 如果安装成功了,你会看到这样结果

    
    Test passed!
    

其实还可以cmake 一下你的caffe/build,也能很快测试是否安装成功

Reference

https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.0.5/prod/Doc/cuDNN-Installation-Guide

One thought on “cudnn Installation in Linux

发表评论

电子邮件地址不会被公开。 必填项已用*标注