如何在 Ubuntu 20.04 上安裝 CUDA

奇蹟 是一種並行計算平台和編程模型，可在利用 GPU 進行日常通用計算時提供卓越的用戶體驗。 CUDA 代表 C計算 你化 D電子設備 一種架構。 它是由英偉達創建的。

CUDA 包括 CUDA 工具包（編譯器、配置文件和調試器）、軟件驅動程序和 CUDA SDK。 軟件層可以直接訪問 GPU 的虛擬指令集和並行計算元素。 對於深度學習研究和框架開發人員使用 神經網絡 用於高性能 GPU 加速。 它是一個用於深度神經網絡的原語庫。

在本指南中，我們將學習如何 在 Ubuntu 20.04 上安裝 CUDA 和 CuDNN. 有兩種方法可以解決這個問題。 您可以從 Ubuntu 存儲庫安裝 CUDA – 這是最簡單的方法 – 或從提供最新版本 CUDA 的 CUDA 存儲庫安裝。

先決條件：

  1. 至少 20 GB 的可用磁盤空間
  2. 推薦 2 GB RAM
  3. 應該安裝了 NVIDIA 圖形驅動程序。 CUDA 附帶的驅動程序可能不是最新版本。

方法 1：從 Ubuntu 存儲庫安裝 CUDA

首先更新 Ubuntu 20.04 實例上的軟件包列表。

$ sudo apt update

接下來，使用 APT 包管理器安裝 CUDA 工具包，如下所示。 請注意，安裝是密集型的，因此請確保您擁有快速穩定的互聯網連接和至少 10G 的磁盤空間。

$ sudo apt install nvidia-cuda-toolkit

安裝完成後，我們需要將 CUDA 添加到 PATH 中，以便通知 shell CUDA 的位置。 為此，我們將在 .bashrc 文件中指定 PATH。 因此，請使用命令行文本編輯器打開該文件。

$ sudo vim ~/.bashrc

在文件的最後，複製並粘貼以下代碼行。

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

接下來，重新加載 .bashrc 文件中的更改。

$ source ~/.bashrc

現在使用以下命令檢查並確認安裝的 CUDA 版本。

$ nvcc --version

檢查 CUDA 版本

方法 2：從 CUDA 存儲庫安裝 CUDA

如前所述，CUDA 存儲庫提供了最新版本的 CUDA。 因此，如果您要安裝最新版本，請按照此處概述的說明進行操作。

首先，下載 CUDA 存儲庫 pin：

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

接下來，將圖釘移到 /etc/preferences.d 目錄並將其重命名為 cuda-repository-pin-600.

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

將 CUDA 引腳移動到 /etc/preferences 目錄將 CUDA 引腳移動到 /etc/preferences 目錄

此後，下載 CUDA Debian 文件。 請注意，該文件非常大 – 在撰寫本指南時準確地說是 2.3G。 確保您有良好的互聯網連接和充足的磁盤空間。

$ wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

下載後，使用 dpkg 實用程序來執行文件。

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

接下來，按如下方式安裝 CUDA GPG 密鑰：

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

最後，更新包列表並使用 APT 包管理器安裝 CUDA。

$ sudo apt update

$ sudo apt install cuda -y

安裝 CuDNN

CuDNN 不附帶 Cuda。 到 下載 CuDNN 您需要註冊才能成為 NVIDIA 開發者計劃的免費會員。

cuDNN linux 下載

在 NVIDIA Developer Program 中註冊後，您可以下載最新版本的 cuDNN。 在這裡，我正在下載適用於 Linux 的 cuDNN 庫。

首先，提取cuDNN包：

$ tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

然後將以下文件複製到CUDA目錄下：

$ 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*

結論

在本教程中，我們學習瞭如何在 Ubuntu 20.04 上安裝 CUDA 和 cuDNN。 感謝您的閱讀，請留下您的反饋和建議。


Related Posts