您的位置:首页 > 其它

ubuntu从零开始安装mxnet--安装cudnn5.0

2017-10-12 18:46 225 查看
在经过之前的各项安装和准备工作《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表示测试成功。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: