天天看點

Ubuntu安裝NVIDIA顯示卡驅動、cuda、cudnn一、Ubuntu安裝NVIDIA顯示卡驅動二、Ubuntu安裝cuda、cudnn

一、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
           
Ubuntu安裝NVIDIA顯示卡驅動、cuda、cudnn一、Ubuntu安裝NVIDIA顯示卡驅動二、Ubuntu安裝cuda、cudnn

3.輸入指令:

sudo apt-get install nvidia-418 nvidia-settings
           

完成後,重新開機Ubuntu

4.檢視顯示卡驅動版本:

nvidia-smi
           
Ubuntu安裝NVIDIA顯示卡驅動、cuda、cudnn一、Ubuntu安裝NVIDIA顯示卡驅動二、Ubuntu安裝cuda、cudnn

二、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
           
Ubuntu安裝NVIDIA顯示卡驅動、cuda、cudnn一、Ubuntu安裝NVIDIA顯示卡驅動二、Ubuntu安裝cuda、cudnn

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
           
Ubuntu安裝NVIDIA顯示卡驅動、cuda、cudnn一、Ubuntu安裝NVIDIA顯示卡驅動二、Ubuntu安裝cuda、cudnn

如果出現上圖所示版本資訊(三個數字的意思是目前版本為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的解除安裝和更新)