如何在 vSphere Bitfusion 中安装 NVIDIA cuDNN

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。

下一步做什么

• 如果使用的是 CentOS 或 Red Hat Linux，则必须先安装 Python 3。请参见如何在 CentOS 和 Red Hat Linux 上安装 Python。
• 如果使用的是 Ubuntu Linux，则可以安装 TensorFlow、PyTorch 和 YOLO。