Skip to content

Latest commit

 

History

10 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 

Repository files navigation

Ubuntu GPU 環境安裝指南


一、安裝 NVIDIA 顯卡驅動

Step 1:禁用 Nouveau

首先檢查 Nouveau 驅動是否在執行:

lsmod | grep nouveau

如果有輸出,表示 Nouveau 正在加載,需要禁用它。編輯黑名單設定文件:

sudo gedit /etc/modprobe.d/blacklist.conf

在文件末尾添加以下內容:

blacklist nouveau
options nouveau modeset=0

保存退出後,重新加載設定並重啟系統:

source ~/.bashrc
reboot

重啟後畫面解析度會降低,這是正常現象。重新檢查是否禁用成功:

lsmod | grep nouveau

若無輸出則表示禁用成功。


Step 2:安裝 NVIDIA 驅動

查看可用的驅動版本:

ubuntu-drivers devices

系統會列出所有可安裝的驅動版本,例如:

  • nvidia-driver-410 (third-party free)
  • nvidia-driver-415 (third-party free)
  • nvidia-driver-450 (third-party free)

安裝前先刪除舊有的版本:

sudo apt-get purge -y 'nvidia-*' 'cuda-*'
sudo apt-get autoremove -y && sudo apt-get autoclean
sudo apt-get update

選擇合適的驅動版本進行安裝(以 570 為例):

sudo apt-get install nvidia-driver-570

驅動安裝完成後,輸入以下命令檢查是否成功:

nvidia-smi

✅ 顯卡驅動安裝完成!


二、安裝 GCC 5 & G++ 5

Ubuntu 18.04 預設使用 GCC 7.0,但 CUDA 9.0 只支持 GCC 6.0 以下版本。

首先檢查目前的 GCC 版本:

gcc --version

如果顯示 7.x 版本,需要降級至 5.x。依次執行以下命令:

sudo apt install gcc-5
sudo apt install g++-5

設定符號連結使系統預設使用 GCC 5:

cd /usr/bin
sudo mv gcc gcc.bak
sudo ln -s gcc-5 gcc
sudo mv g++ g++.bak
sudo ln -s g++-5 g++

驗證版本是否更新:

gcc --version

✅ 應該顯示 5.x 版本


三、安裝 CUDA 11.8

Step 1:下載 CUDA 11.8

  1. 進入 NVIDIA CUDA 官方網站
  2. NVIDIA 預設會導向最新版本,需要手動尋找並點選 CUDA 11.8 版本
  3. 根據作業系統選擇對應的條件進行下載

Step 2:安裝 CUDA

下載完成後,在下載資料夾中右鍵開啟終端機,執行以下命令:

sudo sh cuda_11.8.130_410.48_linux.run

安裝過程中會出現協議文件,可使用空白鍵(SPACE)快速瀏覽。

安裝程序會詢問多個問題:

問題 回答
是否安裝驅動程式 N(因為已先前已安裝)
其他所有問題 Yaccept
路徑相關問題 ENTER 使用預設路徑

✅ CUDA 9.0 安裝完成!

Step 3:設定環境變數

編輯 bashrc 文件:

sudo gedit ~/.bashrc

在文件末尾添加以下內容:

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

保存退出,然後執行:

source ~/.bashrc

驗證 CUDA 是否安裝成功:

nvcc -V

✅ 環境變數設定完成!


四、安裝 cuDNN

下載 cuDNN

  1. 進入 NVIDIA cuDNN 官方網站
  2. 注意: 下載 .tar.xz 格式,不要下載 deb 版本

安裝 cuDNN

解壓縮下載的檔案(將 FileName.tar.xz 替換為實際檔名):

sudo tar Jxvf FileName.tar.xz

進入解壓後的目錄:

cd cuda

複製檔案到 CUDA 安裝目錄:

sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib/libcudnn* /usr/local/cuda/lib64

設定檔案權限:

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

✅ cuDNN 安裝完成!


總結

依照以上步驟順序執行,即可在 Ubuntu 上完成:

  • ✅ NVIDIA 顯卡驅動安裝
  • ✅ GCC/G++ 版本降級
  • ✅ CUDA 11.8 安裝
  • ✅ cuDNN 安裝

整個環境配置完成後,即可開始進行 GPU 相關的開發工作。

About

No description, website, or topics provided.

Resources

Stars

2 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors