nvidia-cudnn
是 NVIDIA CUDA 深度神经网络库(CUDA Deep Neural Network library)的缩写。这是一个由 NVIDIA 提供的库,用于加速深度学习应用程序。它包含了针对深度神经网络中常用操作(如卷积、池化、归一化、激活层等)的高度优化的实现。这些操作都是为了在 NVIDIA 的 GPU 上进行高效计算而特别优化的,从而大大加快深度学习模型的训练和推断速度。cuDNN 是 NVIDIA 深度学习 SDK 的一部分,广泛应用于各种机器学习和深度学习框架中。
- 首先更新您的 APT 数据库:
sudo apt update
- 然后尝试安装
nvidia-cudnn
:
sudo apt -y install nvidia-cudnn
这将安装 NVIDIA CUDA 深度神经网络库(cuDNN)。请注意,这个方法安装的是一个脚本 /usr/sbin/update-nvidia-cudnn
,用于下载和本地安装 NVIDIA cuDNN。这个脚本会在安装过程中自动调用。
安装过程中的图