ubuntu20.0 新系統 安裝N卡驅動、CUDA和CUDNN

前期工作

sudo apt update
sudo apt install openssh-server
sudo apt install vim
sudo apt install net-tools 
           

1、

檢視顯示卡型号

lspci | grep -i nvidia
           

安裝 gcc、g++、make、核

sudo apt-get install build-essential
sudo apt-get install linux-headers-$(uname -r)
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
           

下載下傳2、禁用NOUVEAU

具體操作可參考此部落格

lsmod | grep nouveau
           

若沒有輸出，則為禁用完成。禁用完後重新開機。

3、安裝N卡驅動

sudo chmod a+x NVIDIA-Linux-x86_64-455.45.01.run
sudo ./NVIDIA-Linux-x86_64-455.45.01.run -no-x-check -no-nouveau-check -no-opengl-files
           

安裝完成後，輸入

nvidia-smi

測試，然後重新開機。

4、安裝cuda

sudo sh cuda_11.0.3_450.51.06_linux.run
           

安裝完後

sudo vim ~/.bashrc
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
           

重新整理

source ~/.bashrc

終端中輸入

nvcc -V

，會顯示cuda資訊

5、安裝cudnn

解壓cudnn壓縮包後，

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*