NVIDIA CUDA Deep Neural Network (cuDNN) 是一个 GPU 加速的原语库,用于深度神经网络。

前提条件

  • 创建一个 NVIDIA 开发人员帐户,以从该帐户下载与您的 NVIDIA CUDA 版本匹配并适用于您的 Linux 发行版的 cuDNN 软件包。请参见 https://developer.nvidia.com/cudnn
  • 确认您已安装 vSphere Bitfusion 客户端。
  • 确认您已安装 NVIDIA CUDA。

过程

  1. 通过运行适用于 Linux 分发包的命令序列安装 NVIDIA cuDNN 软件包。
    • Ubuntu 版本 20.04
      sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
    • CentOS 8 和 Red Hat Linux 8
      sudo rpm -ivh libcudnn8-8.0.5.39-1.cuda11.0.x86_64.rpm
  2. (可选) 要验证 NVIDIA cuDNN 是否已安装,请运行 ldconfig -p | grep cudnn

下一步做什么