一、Ubuntu安裝NVIDIA顯示卡驅動
0.删除舊驅動:
sudo apt-get purge nvidia*
1.将ppa:graphics-drivers/ppa存儲庫添加到系統中:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
2.識别顯示卡模型和推薦的驅動程式:
ubuntu-drivers devices

3.輸入指令:
sudo apt-get install nvidia-418 nvidia-settings
完成後,重新開機Ubuntu
4.檢視顯示卡驅動版本:
nvidia-smi
二、Ubuntu安裝cuda、cudnn
0.解除安裝舊cuda
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_10.1.pl
删除cudnn檔案:
sudo rm -rf /usr/local/cuda-10.1
1.安裝cuda
(1)下載下傳cuda_10.1.243_418.87.00_linux.run:
https://developer.nvidia.com/cuda-toolkit-archive
安裝:
sudo sh cuda_10.1.243_418.87.00_linux.run
然後一直按回車,accept ——> n(不要安裝driver)——> y ——> y ——> y
(2)安裝完成後,設定環境變量
主目錄下(/home/sun)打開 .bashrc檔案,添加相關路徑。如果沒有找到,可按Ctrl+H鍵顯示隐藏檔案。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
在終端輸入:
source ~/.bashrc
(3)檢視目前cuda版本
終端輸入:
nvcc --version
3.删除舊cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn
4.安裝新版本的cudnn
(1)下載下傳cudnn-10.1-linux-x64-v7.6.4.38.zip:
https://developer.nvidia.com/rdp/cudnn-archive
解壓檔案,并在解壓出的cuda檔案夾中打開終端,終端輸入:
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
(2)建立軟連結:
終端輸入:
cd /usr/local/cuda/lib64
sudo chmod +r libcudnn.so.7.6.4
sudo ln -sf libcudnn.so.7.6.4 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
(3)檢視cudnn的版本
終端輸入:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出現上圖所示版本資訊(三個數字的意思是目前版本為7.6.4),說明安裝成功。
參看連結:
https://blog.csdn.net/qq_33591712/article/details/94731966 (安裝NVIDIA顯示卡驅動(超簡單))
https://blog.csdn.net/wanzhen4330/article/details/81704474 (Ubuntu16.04下cuda和cudnn的解除安裝和更新)