在经过之前的各项安装和准备工作《ubuntu从零开始安装mxnet–安装NVIDIA驱动》,《ubuntu从零开始安装mxnet–安装cuda8.0》之后。
接下来我们安装cudnn,cudnn的安装比较简单,不过下载可能稍微比较麻烦些。
下载cudnn
我们先要在nvidia的官网注册个账号。然后才能选择对应的cudnn版本下载安装。根据Mxnet官网的要求我们下载5.0的版本。下载下来的内容是个压缩包。
安装cudnn
我们将下载下来的压缩包解压缩,会出来一个cuda文件夹,里面分为include和lib64两个文件夹,我们需要将对应文件夹下的文件拷贝到之前安装cuda8.0的位置下的对应文件夹下。
给个例子,
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/lib* /usr/local/cuda-8.0/lib64
测试安装
在官网中下载cudnn_sample,解压,然后进入
cd cudnn-sample-v5/mnistCUDNN
make
./mnistCUDNN
最后如果出现test passed表示测试成功。