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

在经过之前的各项安装和准备工作《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表示测试成功。
这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值