赞
踩
操作系统:Ubuntun 20.04
1、安装驱动前一定要更新软件列表和安装必要软件、依赖(必须)
sudo apt-get update #更新软件列表
sudo apt-get install g++
sudo apt-get install gcc
sudo apt-get install make
2、查看GPU型号
lspci | grep -i nvidia
3、驱动下载
在nvidia官网(https://www.nvidia.cn/geforce/drivers/)下载一个自己的显卡驱动.
需要注意的是在自己的tensorflow和pytorch需要的cuda版本和cudnn的版本,并下载对应的显卡驱动的版本
4、禁用nouveau
lsmod | grep nouveau
若没有任何输出则说明已经禁用.因为我是重新装,之前禁用过,所以没有任何输出.
若出现一下结果,说明还在运行中
执行以下命令,禁用nouveau。
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
更新并重启。
sudo update-initramfs -u
sudo reboot
重启后输入下面的命令检查是否禁用成功。
若没有任何输出说明禁用 nouveau 驱动成功
lsmod | grep nouveau
5、给驱动程序赋予执行权限
chmod a+x NVIDIA-Linux-*.run
6、安装驱动程序
sudo ./NVIDIA-Linux-*.run -no-x-check -no-nouveau-check -no-opengl-files
7、检查驱动程序是否安装成功
nvidia-smi
文章参考:
https://blog.csdn.net/clfirework/article/details/119028924
https://blog.csdn.net/xianrenli38/article/details/125254853
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。