1 查看显卡型号
ubuntu-drivers devices
1.1 查看GPU型号
lspci | grep -i nvidia
1.2 查看NVIDIA驱动版本
sudo dpkg --list | grep nvidia-*
2 下载显卡驱动
选择手动方式安装驱动的话可以从英伟达官网(https://www.nvidia.com/Download/index.aspx?lang=en-us)进行下载。下载NVIDIA历史版本链接:https://www.nvidia.com/Download/Find.aspx?lang=en-us
其实安装最新版本的驱动并不总是最好的选择,应考虑到对应的cuda适配,深度学习框架对cuda的支持很关键,具体见:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
cuda与cudnn需要满足关系:
https://developer.nvidia.com/rdp/cudnn-archive
cuda历史版本下载连接:
https://developer.nvidia.com/cuda-toolkit-archive
Tensorflow不同版本与CUDA及CUDNN版本对应关系:
https://blog.csdn.net/u013066730/article/details/88845083
3 禁止自带nouveau驱动
1) 卸载已经安装了的nvidia驱动(一般之前没装过的话,也没有nvidia驱动可以卸载)
sudo apt-get remove nvidia-*
sudo apt-get autoremove
或者
sudo apt-get remove --purge nvidia*
# Ubuntu apt-get指令autoclean,clean,autoremove的区别
#具体见:(https://www.jb51.net/article/103101.htm)
2) 禁用自带nouveau驱动
在/etc/modprobe.d/目录下创建blacklist-nouveau.conf文件
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
并添加以下内容
blacklist nouveau
options nouveau modeset=0
然后更新一下
sudo update-initramfs -u
如果更新失败,文件里写
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
更新成功后,重启计算机。
重启计算机后输入命令
lsmod | grep nouveau
没有输出任何东西,说明nouveau禁用成功(一般禁用成功后在本次开机时电脑分辨率会变化,很容易分辨)
4 安装NVIDIA显卡驱动
1) 首先需要安装lightdm,主要是用来关闭启动图形界面用
sudo apt-get install lightdm
2)停止可视化桌面
sudo telinit 3 #可省略
然后 Crtl + Alt + F3 输入用户名和密码后进入命令行
3) 关闭x-Window(不关闭的话驱动无法正常安装)
sudo service lightdm stop
4) 安装NVIDIA驱动
进入到下载的显卡驱动的路径下
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xxx.run #视具体文件而定
sudo ./NVIDIA-Linux-x86_64-xxx.xxx.run --no-opengl-files --no-x-check --no-nouveau-check
## –no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
## –no-x-check 安装驱动时不检查X服务
## –no-nouveau-check 安装驱动时不检查nouveau
安装过程弹出来的选项按原本默认的来选择就行
安装完成后,重启计算机
5)检验是否安装成功
终端输入 nvidia-smi ,有如下显示则表示安装成功