如何在 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。