天天看点

ubuntu 16.04 安装cudnn 5.1

cudnn是nvidia提供的深度学习接口

一、获取cuDNN

官网下载cudnn5.1: https://developer.nvidia.com/rdp/cudnn-download

(需要注册NVIDIA账号)

选择 cuDNN v5.1 -> cuDNN v5.1 Libraryfor Linux

ubuntu 16.04 安装cudnn 5.1

二、安装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
           
ubuntu 16.04 安装cudnn 5.1

文件拷贝成功

ubuntu 16.04 安装cudnn 5.1
ubuntu 16.04 安装cudnn 5.1

查看cudnn版本

sudo cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2
           
ubuntu 16.04 安装cudnn 5.1

继续阅读