服务器设置和教程 · 8 10 月, 2024

Ubuntu 18.04 安裝 GPU + CUDA + cuDNN 以及其版本選擇

Ubuntu 18.04 安裝 GPU + CUDA + cuDNN 以及其版本選擇

在深度學習和高性能計算的領域中，GPU（圖形處理單元）已成為不可或缺的工具。對於使用 Ubuntu 18.04 的開發者來說，安裝 GPU 驅動程式、CUDA 和 cuDNN 是一個重要的步驟。本文將詳細介紹如何在 Ubuntu 18.04 上安裝這些組件，並提供版本選擇的建議。

一、準備工作

在開始安裝之前，請確保您的系統已經更新到最新版本。可以使用以下命令進行更新：

sudo apt update
sudo apt upgrade

接下來，檢查您的 GPU 型號，確保它支持 CUDA。可以使用以下命令來檢查：

lspci | grep -i nvidia

二、安裝 NVIDIA 驅動程式

首先，您需要安裝 NVIDIA 驅動程式。可以通過以下步驟進行安裝：

  1. 添加 NVIDIA PPA：
    2. sudo add-apt-repository ppa:graphics-drivers/ppa
  2. 更新包列表：
    3. sudo apt update
  3. 安裝 NVIDIA 驅動程式：
    4. sudo apt install nvidia-driver-

請將 替換為您所需的驅動版本號。安裝完成後，重啟系統以使驅動生效。

三、安裝 CUDA

接下來，您需要安裝 CUDA。可以從 NVIDIA 的官方網站下載 CUDA 安裝包。以下是安裝步驟：

  1. 訪問 NVIDIA CUDA Toolkit 網站，選擇適合 Ubuntu 18.04 的版本。
  2. 下載 .deb 文件並安裝：
    3. sudo dpkg -i cuda-repo-.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt update
sudo apt install cuda

安裝完成後，您需要將 CUDA 的路徑添加到環境變量中。可以編輯 ~/.bashrc 文件，添加以下行：

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然後運行以下命令使更改生效：

source ~/.bashrc

四、安裝 cuDNN

cuDNN 是 NVIDIA 提供的深度學習加速庫。要安裝 cuDNN，您需要從 NVIDIA 開發者網站下載相應版本的 cuDNN。以下是安裝步驟：

  1. 訪問 NVIDIA cuDNN 網站，選擇與您安裝的 CUDA 版本相對應的 cuDNN 版本。
  2. 下載 cuDNN 的 tar 文件並解壓：
    3. tar -xzvf cudnn--linux-x64-v.tgz
  3. 將解壓後的文件複製到 CUDA 目錄：
    4. sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

五、版本選擇建議

在選擇 CUDA 和 cuDNN 的版本時，建議遵循以下原則：

  • 確保 CUDA 和 cuDNN 的版本相互兼容。可以參考 NVIDIA 的官方文檔來確認版本兼容性。
  • 選擇穩定版本而非測試版本，以確保系統的穩定性和性能。
  • 根據您的深度學習框架（如 TensorFlow 或 PyTorch）來選擇合適的版本，因為不同框架對 CUDA 和 cuDNN 的支持版本可能有所不同。

總結

在 Ubuntu 18.04 上安裝 GPU、CUDA 和 cuDNN 是一個相對直接的過程，但需要注意版本的兼容性和正確的安裝步驟。通過遵循上述指南，您可以為深度學習和高性能計算搭建一個強大的環境。如果您需要更高效的計算資源，可以考慮使用我們的香港VPS服務，為您的項目提供穩定的支持。

You may also like...