Ubuntu 18.04安装CUDA 11.4.0 cuDNN 8.2.2

最新推荐文章于 2026-04-24 02:01:35 发布

原创

最新推荐文章于 2026-04-24 02:01:35 发布 · 4.8k 阅读

标签

#ubuntu #深度学习 #人工智能

收录于

本文介绍了如何在主机上安装CUDA和cuDNN以支持GPU加速的神经网络训练和推理。首先确认NVIDIA显卡驱动无误，然后通过nvidia-smi检查适用的CUDA版本。接着从NVIDIA官网下载对应版本的CUDA和cuDNN，按照步骤进行安装，并验证安装是否成功。最后，将cuDNN的库文件和头文件复制到CUDA目录完成安装。

CUDA和cuDNN为NVIDIA支持GPU运算以及深度神经网络计算加速的算法库。通常需要安装以支持利用GPU加速神经网络的训练和推理。

安装前需要确定主机显卡为NVIDIA显卡，且驱动安装无误。通过nvidia-smi查看显卡信息和适合的CUDA版本。

$ nvidia-smi

不同的显卡需要下载安装对应的CUDA与cuDNN版本，安装前需前往Navida官网查看版本信息。

CUDA版本下载： https://developer.nvidia.com/cuda-toolkit-archive
cuDNN版本下载：https://developer.nvidia.com/rdp/cudnn-archive

个人主机显卡驱动：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.129.06   Driver Version: 470.129.06   CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00