1 顯卡驅(qū)動
- 可以在N卡官網(wǎng)尋找驅(qū)動 > Download Drivers
- 或者直接使用apt安裝，先選擇我們需要安裝的顯卡驅(qū)動，輸入命令
ubuntu-drivers devices
結(jié)果顯示：
新版本的Ubuntu不需要nouveau等操作句狼，直接按上面建議的安裝（可以看到有一個recommened）
sudo apt install nvidia-driver-450
或者：
sudo ubuntu-drivers autoinstall
我在安裝后出現(xiàn)如下報錯扒怖，重啟即可
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.
sudo reboot
其他情況下這樣的報錯可以試試這個方法：
sudo apt-get install dkms
sudo dkms install -m nvidia -v XXX.XX
2 安裝CUDA
- 先去官網(wǎng)下載對應的包埋凯，選Ubuntu18.04之后的版本即可 > CUDA Toolkit Archive
這里提一句CUDA和顯卡驅(qū)動的對應關系狐蜕，安裝CUDA必須要有最低某版本的顯卡驅(qū)動煞聪，而不是一一對應關系雾棺。也就是說理論上我安裝了最新的驅(qū)動可以安裝任意版本的CUDA膊夹，看你的需求來定（比如tensorflow, pytorch等需要和CUDA版本對應）。比如我圖中選擇安裝的CUDA10.1捌浩，需要最低418.39的驅(qū)動版本號放刨。對于我們安裝最新驅(qū)動的情況自然不成問題。
- 由于我們的系統(tǒng)版本是20.04尸饺，自帶gcc-9进统，需要降級才能完成安裝助币，降級方法參考http://www.reibang.com/p/a508a6e0e0b7
- 安裝
chmod +x cuda_10.1.105_418.39_linux.run
sudo ./cuda_10.1.105_418.39_linux.run
然后accept，注意下面把Driver去掉（我們已經(jīng)有driver并不想換）螟碎，只裝CUDA Toolkit即可眉菱，然后Install
安裝完畢，這里提示我們把環(huán)境加到系統(tǒng)變量里掉分，我們在~/.bashrc的末尾添加如下兩行俭缓，并source生效：
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
使用nvcc -V檢驗是否安裝成功
3 安裝cuDNN
到官網(wǎng)登錄，下載對應版本的cuDNN > cuDNN download
得到 cudnn-10.1-linux-x64-v8.0.4.30.solitairetheme8：
mv cudnn-10.1-linux-x64-v8.0.4.30.solitairetheme8 cudnn.tar
tar -xvf cudnn.tar
sudo cp ./cuda/include/* /usr/local/cuda/include/
sudo cp ./cuda/lib64/* /usr/local/cuda/lib64/
至此cuDNN完成安裝
