-
查看GPU版本
lspci | grep -i nvidia
-
从下面的网址ThePCI ID Repository中输入ID查看对应的GPU版本
-
官网NVIDIA下载对应的驱动
-
安装下载的文件
sudo sh ./NVIDIA-Linux-x86_64-535.86.05.run
-
检验是否安装成功
nvidia-smi
-
CUDA安装,官网
-
设置环境变量
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}}
-
检测是否安装成功
-
安装cuDNN,官网