ubuntu 下安装 cudnn

安装cudnn前后,GPU跑一个算法的速度分别是139ms和26ms !

1. 在以下网址选择  cuDNN v5.1 Library for Linux 下载

https://developer.nvidia.com/rdp/cudnn-download

2.下载得到一个 cudnn-8.0-linux-x64-v5.1.tgz 包,解压之

得到 cuda 文件夹, 下含两个子文件夹 include , lib64

3. 把 include 和 lib64 文件夹内容分别复制到

/usr/local/cuda/include

/usr/local/cuda/lib64 

原文地址:https://www.cnblogs.com/rabitvision/p/6337973.html