GPU显卡驱动安装

• 查看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，官网
  

参考资料
linux版本安装cuda
Linux系统（Ubuntu）下安装及配置CUDA过程详解