如何在 Ubuntu 18.04 上安裝 CUDA

CUDA 旨在通過利用系統上圖形處理單元 (GPU) 的強大功能來顯著提高計算性能。 在本文中,我將向您展示如何在 Ubuntu 18.04 上安裝 CUDA。 CUDA 是英偉達開發的並行編程模型和計算平台。

您可以使用以下方法之一在 Ubuntu 18.04 上安裝 CUDA:

  1. 來自與分發無關的包(運行文件包)
  2. 來自特定於發行版的包(RPM 和 Deb 包)

CUDA 預安裝檢查

以下是在 Ubuntu 系統上安裝 CUDA 工具包和驅動程序之前需要執行的一些檢查。

驗證系統具有支持 CUDA 的 GPU

您需要驗證您的 GPU 是否可以與 CUDA 配合使用,運行以下命令進行檢查:

$ lspci | grep -i nvidia
01:00.0 3D controller: NVIDIA Corporation GM206M [GeForce GTX 965M] (rev a1)

如果您剛剛安裝了驅動卡,您可能需要手動更新上述命令的 PCI 數據庫以返回有效輸出。

# update-pciids

驗證系統是否安裝了 gcc

對於使用 CUDA 的開發,您需要確保安裝了 gcc。 您可以使用以下命令檢查它是否已安裝:

$ gcc --version

如果未安裝,請使用 apt-get 安裝,如下所示:

$ sudo apt install gcc-6 g++-6

驗證系統安裝了正確的內核頭文件和開發包。

CUDA 驅動程序要求在安裝驅動程序時以及在重新構建驅動程序時安裝用於運行內核版本的內核頭文件和開發包。 您可以使用以下方法安裝內核頭文件和開發工具:

$ sudo apt-get install linux-headers-$(uname -r)

安裝英偉達驅動

CUDA 需要在您的機器上安裝 Nvidia 驅動程序。 使用以下命令在 Ubuntu 18.04 上安裝它:

$ sudo apt install nvidia-384

安裝完成後,您可以繼續安裝 Nvidia CUDA 工具包。

下載 NVIDIA CUDA 工具包

根據您選擇的安裝方法,您需要下載等效的軟件包。 我更喜歡從 Ubuntu 18.04 上的運行文件安裝 CUDA,因為它很難遇到依賴問題。

在撰寫本文時,CUDA 的最新版本是 v9.2。 由於包大小超過 1GB,我將使用 wget 命令下載它,以便在連接中斷時可以輕鬆恢復。 CUDA 工具包包含創建、構建和運行 CUDA 應用程序所需的 CUDA 驅動程序和工具,以及庫、頭文件、CUDA 示例源代碼和其他資源。

$ cd Dowloads
$ wget -c https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda_9.2.88_396.26_linux

將包下載到本地後,使其可執行並安裝。

# chmod +x cuda_9.2.88_396.26_linux.run
# ./cuda_9.2.88_396.26_linux.run --verbose --silent --toolkit --override

完成安裝後,您應該會得到類似於下面的輸出。

===========
= Summary =
===========

Toolkit: Installed in /usr/local/cuda-9.2
Samples: Not Selected

Please make sure that
 - PATH includes /usr/local/cuda-9.2/bin
 - LD_LIBRARY_PATH includes /usr/local/cuda-9.2/lib64, or, add /usr/local/cuda-9.2/lib64 to /etc/ld.so.conf and run ldconfig as root

修改你的 .bashrc 文件以在其路徑中包含 Cuda bin:

export PATH="$PATH:/usr/local/cuda-9.2/bin"

確保存在 CUDA 庫路徑。

# echo "/usr/local/cuda-9.2/lib64" >> /etc/ld.so.conf
# ldconfig

檢查 https://developer.nvidia.com/cuda-downloads 以獲取可用補丁並下載 。跑 文件然後安裝它。

# wget https://developer.nvidia.com/compute/cuda/9.2/Prod/patches/1/cuda_9.2.88.1_linux
# chmod +x cuda_9.2.88.1_linux.run
# ./cuda_9.2.88.1_linux.run --silent --accept-eula

Welcome to the CUDA Patcher.
Installation complete!
Installation directory: /usr/local/cuda-9.2

在 CUDA bin 文件夾中創建 GCC6 的符號鏈接:

# ln -s /usr/bin/gcc-6 /usr/local/cuda-9.2/bin/gcc
# ln -s /usr/bin/g++-6 /usr/local/cuda-9.2/bin/g++

另請閱讀:

  • 如何在 Ubuntu 18.04 上安裝 Nvidia 驅動程序

您現在應該已經準備好了 CUDA 工具包。 開始創建您的項目並享受它。