为深度学习所用,博主预想在Ubuntu16.04上安装 显卡驱动 + CUDA + cuDNN + Tensorflow-gpu + Keras + PyCharm,参考了众多资料,最终成功将所有软件安装完毕,且能成功运行使用。该篇博客介绍了显卡驱动的安装教程,亦可通过下方链接查看其他步骤的安装教程。
换源及安装显卡驱动
安装 Tensorflow-gpu 与 Keras(附测试)
换源
该步骤的目的是为下载显卡驱动做好准备,若不换源,显卡驱动将下载得特别特别特别慢!!!
1、备份源,将sources.list备份到sources.list.bak
cd /etc/apt/
sudo cp sources.list sources.list.bak
2、换源,换成阿里云源,打开sources.list文件后,直接将以下的阿里云源替换原本文件的全部内容
sudo gedit sources.list
阿里云源:
# deb cdrom:[Ubuntu 16.04 LTS _Xenial Xerus_ - Release amd64 (20160420.1)]/ xenial main restricted
deb-src http://archive.ubuntu.com/ubuntu xenial main restricted #Added by software-properties
deb http://mirrors.aliyun.com/ubuntu/ xenial main restricted
deb-src http://mirrors.aliyun.com/ubuntu/ xenial main restricted multiverse universe #Added by software-properties
deb http://mirrors.aliyun.com/ubuntu/ xenial-updates main restricted
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates main restricted multiverse universe #Added by software-properties
deb http://mirrors.aliyun.com/ubuntu/ xenial universe
deb http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
deb http://mirrors.aliyun.com/ubuntu/ xenial multiverse
deb http://mirrors.aliyun.com/ubuntu/ xenial-updates multiverse
deb http://mirrors.aliyun.com/ubuntu/ xenial-backports main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-backports main restricted universe multiverse #Added by software-properties
deb http://archive.canonical.com/ubuntu xenial partner
deb-src http://archive.canonical.com/ubuntu xenial partner
deb http://mirrors.aliyun.com/ubuntu/ xenial-security main restricted
deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security main restricted multiverse universe #Added by software-properties
deb http://mirrors.aliyun.com/ubuntu/ xenial-security universe
deb http://mirrors.aliyun.com/ubuntu/ xenial-security multiverse
3、保存并关闭sources.list文件后,使用以下命令更新源
sudo apt-get update
4、查看源的更新情况,在Ubuntu系统最上方的菜单栏的最右侧,有个关机按钮,选中其中的“系统设置…”,在弹出界面的“系统”中选中“软件和更新”,在最新的弹出界面中可查看目前使用的源地址
安装显卡驱动
安装显卡驱动的方式共有三种:
(1)使用Ubuntu自带的驱动进行安装
(2)使用PPA源进行安装
(3)使用官方的Nvidia驱动包进行手动安装
由于Ubuntu自带的驱动可能不满足所需,而官方的Nvidia驱动包安装过于复杂,容易出错,因此本教程选择使用PPA源的安装方法,精简显卡驱动的安装过程。若采用本教程的方法未能成功安装显卡驱动,可自行搜索使用官方的Nvidia驱动包进行手动安装的复杂方法。
1、禁用 Nouveau,防止显卡驱动安装完毕,电脑重启后循环停留在用户登录界面的情况
在终端使用以下命令查看此时nouveau的加载情况,若没有输出,则说明nouveau已被禁用,可跳过该步骤,若有输出,则需要禁用nouveau
lsmod | grep nouveau
在 /etc/modprobe.d 中创建 blacklist-nouveau.conf 文件
sudo vi /etc/modprobe.d/blacklist-nouveau.conf
在 blacklist-nouveau.conf 文件中输入以下内容(键入“i”转成输入模式,将内容输入后,键入“Esc”退出输入模式,而后键入“:wq”保存并退出文件)
blacklist nouveau
options nouveau modeset=0
更新文件
sudo update-initramfs -u
重启
sudo reboot
重启后再次输入以下命令,若无内容输出,则nouveau已被禁用
lsmod | grep nouveau
2、添加PPA源
sudo add-apt-repository ppa:graphics-drivers/ppa
3、更新系统
sudo apt-get update
4、查看可供安装的显卡驱动版本
ubuntu-drivers devices
也可在 系统设置–>软件和更新–>附加驱动 中查看
5、根据版本,安装所需的显卡驱动(xxx代表驱动版本号)
sudo apt-get install nvidia-driver-xxx
也可使用图形化界面,安装所需版本的显卡驱动,选择需要安装的显卡驱动,点击“应用更改”,等待安装完毕
6、安装完成后,重启电脑
sudo reboot
7、重启后,在终端输入以下命令,查看此时的显卡驱动情况
nvidia-smi
也可通过 系统设置–>详细信息 查看此时使用的显卡