安装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
时间: 2024-10-11 00:35:59