安裝 CUDA Toolkit 11.7

安裝 CUDA 驅動的方法很多可以在官網上的安裝網頁看到 CUDA Toolkit 11.7 Downloads 其中包括

  • deb (local):離線安裝模式
  • deb (network):線上安裝模式
  • runfile (local):離線安裝腳本模式

以上方法都可以成功安裝案需求使用,不過我個人覺得使用 runfile (local) 會比較好安裝,它再安裝的過程中有小小的人機對話介面可以讓安裝的步驟比較不會這麼困難。

使用網路線上安裝

線上安裝方法就是把 NVIDA 的 apt 位置加入系統中再使用 apt 安裝它,通常這樣會比較推薦因為它可以跟著系統更新一起更新。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

使用 CUDA 安裝腳本安裝

使用此方法會下載含有 CUDA 離線安裝腳本執行後會看到對話介面只要依照它的步驟安裝通常就可以正常安裝成功。

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

離線安裝

如果你的系統沒辦法連網會推薦使用此方法安裝,因為它可以直接把檔案下載後放入無法上網的電腦離線安裝 CUDA 驅動程式。

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get updatesudo 
apt-get -y install cuda

參考資料