Ubuntu 16.04 安裝 CUDA 8.0

CUDA（Compute Unified Device Architecture）是由 NVIDIA 開發的一種平行計算平台和程式設計模型，旨在利用 NVIDIA 顯示卡的強大計算能力。對於需要進行高效能計算的開發者和研究人員來說，安裝 CUDA 是一個重要的步驟。本文將指導您如何在 Ubuntu 16.04 上安裝 CUDA 8.0。

系統要求

在安裝 CUDA 8.0 之前，請確保您的系統符合以下要求：

• 操作系統：Ubuntu 16.04
• 支援的 NVIDIA 顯示卡
• 適當的驅動程式版本（至少 367.48 以上）

步驟一：更新系統

在開始安裝之前，建議您更新系統以確保所有套件都是最新的。可以使用以下命令：

sudo apt-get update
sudo apt-get upgrade

步驟二：安裝 NVIDIA 驅動程式

CUDA 需要 NVIDIA 驅動程式的支持。您可以通過以下命令安裝最新的驅動程式：

sudo apt-get install nvidia-367

安裝完成後，重啟系統以使驅動程式生效：

sudo reboot

步驟三：下載 CUDA 8.0 安裝包

前往 NVIDIA 的官方網站，下載 CUDA 8.0 的安裝包。您可以使用以下命令直接下載：

wget https://developer.download.nvidia.com/compute/cuda/8.0/Prod/local_installers/cuda-repo-ubuntu1604-8-0-local_8.0.61-1_amd64-deb

步驟四：安裝 CUDA 8.0

下載完成後，您需要安裝 CUDA 的本地存儲庫。使用以下命令：

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.61-1_amd64-deb

接下來，添加 GPG 密鑰：

sudo apt-key add /var/cuda-repo-ubuntu1604-8-0-local/7fa2af80.pub

然後更新存儲庫並安裝 CUDA：

sudo apt-get update
sudo apt-get install cuda

步驟五：設置環境變數

安裝完成後，您需要設置環境變數以便系統能夠找到 CUDA。編輯您的 .bashrc 文件：

nano ~/.bashrc

在文件末尾添加以下行：

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存並退出編輯器，然後運行以下命令以使更改生效：

source ~/.bashrc

步驟六：驗證安裝

要驗證 CUDA 是否安裝成功，可以運行以下命令：

nvcc -V

如果顯示 CUDA 8.0 的版本信息，則表示安裝成功。

總結

在 Ubuntu 16.04 上安裝 CUDA 8.0 的過程相對簡單，只需遵循上述步驟即可完成。這將使您能夠利用 NVIDIA 顯示卡的計算能力，進行高效能計算和深度學習等應用。如果您需要穩定的伺服器環境來運行您的應用，考慮使用 香港VPS 服務，以獲得更好的性能和可靠性。