Ubuntu20.04配置显卡驱动

https://www.nvidia.cn/geforce/drivers/

ubuntu-drivers devices

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



nvidia-smi

安装cuda

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

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


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

记得cuda toolkit版本号

sudo vim ~/.bashrc


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

cudaNN
https://developer.nvidia.com/

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

检查

 nvcc -V