Install CuDNN on Ubuntu 16.04?

最新推荐文章于 2024-10-14 23:04:42 发布

转载最新推荐文章于 2024-10-14 23:04:42 发布 · 3k 阅读

计算机专栏收录该内容

93 篇文章

订阅专栏

本文介绍如何在 Ubuntu 16.04 上安装 CUDA 和 CuDNN，以支持 TensorFlow 或 Theano 的运行。步骤包括从官方仓库安装 CUDA，注册 NVIDIA 开发者账号下载 CuDNN，并将 CuDNN 库文件复制到正确的位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

摘自http://askubuntu.com/questions/767269/how-can-i-install-cudnn-on-ubuntu-16-04

For TensorFlow or theano, I would like to install cuda and CuDNN. How do I do that on Ubuntu 16.04?

Answer:

Step 0: Install cuda from the standard repositories.

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 /urs/local/cuda/. You can check it with which nvcc or ldconfig -p | grep cuda

Step 3: Copy the files:

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