如何安装cuDNN

随笔 2016-01-05

  cuDNN是英伟达针对CNN(卷积神经网络)推出的一个库,据说可以对CNN过程进行加速,既然这么好,不妨试用一下。下载地址是链接,需要先注册,然后按步骤选择平台下载,现在有v3和v4版本,可以根据你的CUDA版本选择。我选择的是cuDNN v4 Library for Windows。当然也有一些Sample可以下载。   下载后解压得到一个cuda文件夹,可以先将它放到任意位置,比如我放在D盘根目录。   1.然后设置环境变量,将D:\cuda\bin(取决于你的文件位置)添加到系统变量的Path中。   2.用VS新建一个cuda项目(前提是你的CUDA配置完成)。   3.设置项目属性:

1.png

   a.配置设为Release,平台设为x64   b.VC++目录——>包含目录 添加cuda文件的include路径,我的是D:\cuda\include   c.VC++目录——>库目录 加上cuda文件的x64路径 2.png

  d.连接器——>输入——>附加依赖项 添加 cudnn.lib   e.CUDA C/C++ ——>Device 添加 compute_20,sm_30

  4.新建一个cpp文件,添加以下代码:

  #include <iostream>
#include <cuda_runtime.h>
#include <cudnn.h>
using namespace std;

void main(){
cudnnHandle_t handle;
cudnnStatus_t t = cudnnCreate(&handle);
cout<< cudnnGetErrorString(t);
getchar();
}

  此时应该不会有报错。

  运行后输出 3.png

  表示安装成功。

  由于官方资料不足,特补上此文,希望对大家安装cuDNN有所帮助。


本文由 Tony 创作,采用 知识共享署名 3.0,可自由转载、引用,但需署名作者且注明文章出处。

如果对您有用,您的支持将鼓励我继续创作!