安装NVIDIA驱动
sudo apt update
下载依赖
sudo apt install gcc make vim ssh -y
查看是否禁用nouveau
lsmod | grep nouveau
显示输出代表没禁用
禁用自带的nouveau开源驱动
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
文本输入以下两行
blacklist nouveau
options nouveau modeset=0
更新
sudo update-initramfs -u
重启
reboot
再次查看是否禁用nouveau
lsmod | grep nouveau
没有输出代表禁用成功
下载NVIDIA驱动
wget https://us.download.nvidia.com/XFree86/Linux-x86_64/550.107.02/NVIDIA-Linux-x86_64-550.107.02.run
下载完成赋权
sudo chmod a+x NVIDIA-Linux-x86_64-550.107.02.run
安装
sudo ./NVIDIA-Linux-x86_64-550.107.02.run -no-x-check -no-opengl-files
–no-opengl-files 只安装驱动文件,不安装OpenGL文件 这个参数很重要,不加很有可能出现循环登录,也就是loop login
–no-nouveau-check 安装驱动时不检查nouveau
–no-x-check 安装驱动时不检查X服务
提示nvidia-xconfig 选yes表示独显输出 选no表示集显输出
安装cuda
12.4.1
下载
wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run
安装
sudo sh cuda_12.4.1_550.54.15_linux.run
12.6.1
下载
wget https://developer.download.nvidia.com/compute/cuda/12.6.1/local_installers/cuda_12.6.1_560.35.03_linux.run
安装
sudo sh cuda_12.6.1_560.35.03_linux.run
添加环境变量
sudo vim ~/.bashrc
在文末添加下两行并保存关闭
export PATH=/usr/local/cuda-12.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}