cudnn是nvidia提供的深度学习接口
一、获取cuDNN
官网下载cudnn5.1: https://developer.nvidia.com/rdp/cudnn-download
(需要注册NVIDIA账号)
选择 cuDNN v5.1 -> cuDNN v5.1 Libraryfor Linux
![](https://img.laitimes.com/img/_0nNw4CM6IyYiwiM6ICdiwiI0NXYFhGd192UvwVe0lmdhJ3ZvwFM38CXlZHbvN3cpR2Lc1TPB10QGtWUCpEMJ9CXsxWam9CXwADNvwVZ6l2c052bm9CXUJDT1wkNhVzLcRnbvZ2Lc1zdyEWNW1mYshmMZZXUYpVd1kmYr50MZV3YyI2cKJDT29GRjBjUIF2LcRHelR3LcJzLctmch1mclRXY39zNzcTM1cjM2EDOyEDM4EDMy8CX0Vmbu4GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.jpg)
二、安装cuDNN
把头文件和库文件拷贝到cuda的安装目录中,首先确保cuda的安装目录是/usr/local/cuda/,之前安装cuda时,已经默认安装目录为/usr/local/cuda/了
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
文件拷贝成功
查看cudnn版本
sudo cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2