cudnn Installation in Linux
cudnn Installation in Linux
安装教程相对简单,实际原理就是将 cudnn的库加入 cuda 的 include 和 lib 文件夹中
0.前提
- 此方法适用于Linux 系统,常见的
Ubuntu,Debian等等 - 首先你的系统已经成功安装好
CUDA toolkit - 从官方网站上下载
cudnn的压缩包(很多版本,下载你需要的),需要登录账户
1. Ubuntu 下安装
-
进入存在 cudnn**.tgz 的文件夹
-
解压 ``` $ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
1
2
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*
1
2
3
4
5
6
7
8
## 2\. `Debian` 下安装
`Debian`下的安装方法跟简单,直接安装打包好的`.deb`就可以了.
1. 进入`debian`版本`cudnn`的文件夹下
2. 安装运行时的库
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
1
2
3. 安装开发者的库
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
1
2
4. 安装 例程和用户指南
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
1
2
3
4
5
6
7
8
## 3.测试一下是否安装成功
运行一个小Demo即可.
如果安装了 例程和用户指南 这个包的话,我们可以找到位于 `/usr/src/cudnn_samples_v7`的`mnistCUDNN`这个小例子.
1. 拷贝到 你的home/yourdir 任意文件夹下
$cp -r /usr/src/cudnn_samples_v7/ $HOME
1
2
2. 进入 `mnistCUDNN`
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
1
2
3. 编译
$make clean && make
1
2
4. 运行
$ ./mnistCUDNN
1
2
5. 如果安装成功了,你会看到这样结果
Test passed!
```
其实还可以cmake 一下你的caffe/build,也能很快测试是否安装成功