Ubuntu20.04配置显卡驱动

（一）配置显卡驱动：

先选择我们需要安装的显卡驱动

ubuntu-drivers devices

没有就需要安装，再继续上一步

sudo apt ubuntu-drivers devices

安装推荐版本

sudo apt install nvidia-driver-535-server-open

出现以下问题：

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

查看显卡驱动：

nvidia-smi

(二）安装cuda

到nvidia官网下载对应的cuda工具包

https://developer.nvidia.com/cuda-toolkit-archive

选择对应的cuda版本，查看显卡时会显示

下载对应的版本：

下载：

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.ru

sudo sh cuda_12.2.0_535.54.03_linux.run

accept

把Driver去掉（我们已经有driver并不想换），只装CUDA Toolkit即可，然后Install

记得cuda toolkit版本号

安装完毕，这里提示我们把环境加到系统变量里，我们在~/.bashrc的末尾添加如下两行，并source生效：

sudo vim ~/.bashrc

按i编辑

export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

退出保存 

source ~/.bashrc

使用nvcc -V检验是否安装成功 

 

(三）安装cuDNN

https://developer.nvidia.com/ 

到官网注册下载

解压完,把下面两个文件分别放入cuda

sudo cp ./include/* /usr/local/cuda/include/
sudo cp ./lib/* /usr/local/cuda/lib64/

cuDNN完成安装！