- 下载驱动和卸载、禁用自带驱动程序
1.1 查看系统显卡型号
lspci | grep -i nvidia
1
1.2 从NVIDIA官网下载相应驱动
官网网址:https://www.nvidia.com/Download/index.aspx?lang=en-us
1.3 卸载Ubuntu自带的驱动程序
sudo apt purge nvidia*
1
1.4 禁用自带的nouveau nvidia驱动
打开/etc/modprobe.d/blacklist.conf
sudo vi /etc/modprobe.d/blacklist.conf
1
如果提示没有vi就执行下面命令安装一下:
sudo apt install vi
1
在blacklist.conf文件中最后添加如下内容
blacklist nouveau
options nouveau modeset=0
1
2
然后保存退出。
1.5 更新
sudo update-initramfs -u
1
1.6 重启电脑
sudo reboot
1
1.7 查看是否将自带的驱动屏蔽
lsmod | grep nouveau
1
没有结果输出,则表示屏蔽成功。
- 安装显卡驱动
2.1 停止lightdm桌面服务
sudo service lightdm stop
1
如果提示没有lightdm,使用下面命令安装,
sudo apt install -y lightdm
1
再执行上面的命令。如果按照的服务器版本的ubuntu,是没有图像化桌面的,可以忽略2.1步骤。(注意,服务器版的ubuntu不需要这一步;如果是桌面版的ubuntu,需要这一步,最后安装完成之后需要命令重新启动桌面服务,不然没有图形化界面,只有终端形式,感谢用户小皮每天进步一点点的评论。)
2.2 安装驱动
2.2.1 首先查看你有没有安装gcc
gcc --version
1
如果没有安装gcc一般make也没安装,这时需要使用下面命令安装一下gcc和make
【需要将gcc升级到12版本以上)
2.2.2 执行安装
cd到你下载的显卡驱动的目录
sudo chmod a+x NVIDIA-Linux-x86_64-525.89.02.run
sudo ./NVIDIA-Linux-x86_64-525.89.02.run -no-x-check -no-nouveau-check -no-opengl-files
1
2
-no-x-check: 安装时关闭X服务;
-no-nouveau-check: 安装时禁用nouveau;
-no-opengl-files: 只安装驱动文件,不安装OpenGL文件。
安装过程中下面步骤按照下图选择,其它默认。
安装完成以后执行
nvidia-smi
1
如果显示如下内容则表示安装成功。
3 安装cuda
内容在:https://blog.csdn.net/qq_49323609/article/details/130310522
root@aplomb-System-Product-Name:~# nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Thu_Nov_18_09:45:30_PST_2021 Cuda compilation tools, release 11.5, V11.5.119 Build cuda_11.5.r11.5/compiler.30672275_0
root@aplomb-System-Product-Name:~# nvidia-smi Thu Jun 27 17:59:06 2024 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 550.90.07 Driver Version: 550.90.07 CUDA Version: 12.4 |
版本不一致
sudo apt-get --purge remove "*cublas*" "cuda*" "nsight*"
sudo apt-get autoremove
sudo apt-get autoclean
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run
会提示卸载驱动程序重新安装,有风险
采取下面的安装方法
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4
设置环境变量: 安装完成后,更新你的环境变量以指向新的 CUDA 版本。编辑你的 .bashrc
文件:
nano ~/.bashrc
添加以下行到文件末尾:
export PATH=/usr/local/cuda-12.4/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH
保存并关闭文件,然后运行以下命令使更改生效:
source ~/.bashrc
验证安装: 运行以下命令来验证
nvcc
是否已经升级到 12.4 版本:nvcc --version
你应该看到
nvcc
的版本信息显示为 12.4。