首先檢查 Nouveau 驅動是否在執行:
lsmod | grep nouveau如果有輸出,表示 Nouveau 正在加載,需要禁用它。編輯黑名單設定文件:
sudo gedit /etc/modprobe.d/blacklist.conf在文件末尾添加以下內容:
blacklist nouveau
options nouveau modeset=0
保存退出後,重新加載設定並重啟系統:
source ~/.bashrc
reboot重啟後畫面解析度會降低,這是正常現象。重新檢查是否禁用成功:
lsmod | grep nouveau若無輸出則表示禁用成功。
查看可用的驅動版本:
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✅ 顯卡驅動安裝完成!
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 版本
- 進入 NVIDIA CUDA 官方網站
- NVIDIA 預設會導向最新版本,需要手動尋找並點選 CUDA 11.8 版本
- 根據作業系統選擇對應的條件進行下載
下載完成後,在下載資料夾中右鍵開啟終端機,執行以下命令:
sudo sh cuda_11.8.130_410.48_linux.run安裝過程中會出現協議文件,可使用空白鍵(SPACE)快速瀏覽。
安裝程序會詢問多個問題:
| 問題 | 回答 |
|---|---|
| 是否安裝驅動程式 | N(因為已先前已安裝) |
| 其他所有問題 | Y 或 accept |
| 路徑相關問題 | 按 ENTER 使用預設路徑 |
✅ CUDA 9.0 安裝完成!
編輯 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✅ 環境變數設定完成!
- 進入 NVIDIA cuDNN 官方網站
- 注意: 下載
.tar.xz格式,不要下載 deb 版本
解壓縮下載的檔案(將 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 相關的開發工作。