Install CuDNN on Ubuntu 16.04

最新推荐文章于 2025-04-02 14:58:42 发布

转载最新推荐文章于 2025-04-02 14:58:42 发布 · 262 阅读

AI/ML/DL 专栏收录该内容

254 篇文章

订阅专栏

本文详细介绍了如何在Ubuntu 16.04上安装CUDA和cuDNN的步骤。首先，从标准仓库安装CUDA，然后注册NVIDIA开发者账户并下载cuDNN。接着检查CUDA安装位置，并根据不同的安装方式复制必要的文件到指定目录，最后调整权限确保所有用户可以读取cuDNN库。

From: https://askubuntu.com/questions/767269/how-can-i-install-cudnn-on-ubuntu-16-04

129

Step 0: Install cuda from the standard repositories. (See How can I install CUDA on Ubuntu 16.04?)

Step 1: Register an nvidia developer account and download cudnn here (about 80 MB)

Step 2: Check where your cuda installation is. For the installation from the repository it is /usr/lib/... and /usr/include. Otherwise, it will be /usr/local/cuda/ or /usr/local/cuda-<version>. You can check it with which nvcc or ldconfig -p | grep cuda

Step 3: Copy the files:

Repository installation:

$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

Runfile installation:

$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

For Tensorflow to find everything, I had to copy include/cudnn.h and the libraries in lib64/ to /usr/local/cuda-8.0/include and /usr/local/cuda-8.0/lib64 (using CUDA 8.0, Ubuntu 14.04, Tensorflow 0.12.0rc0) - maybe this is helpful for somebody. – David Stutz