欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu 16.04.05 LTS + cuDNN 5.1安装教程

程序员文章站 2022-06-05 22:29:42
...

Ubuntu 16.04.05 LTS + cuDNN 5.1安装教程

一、获取cuDNN

cuDNN与CUDA对应版本查询(https://developer.nvidia.com/rdp/cudnn-archive)(下载需要注册NVIDIA账号)

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

Ubuntu 16.04.05 LTS + 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

查看cudnn版本

Ubuntu 16.04.05 LTS + cuDNN 5.1安装教程