深度学习环境搭建——安装GPU驱动
1.1 首先使用下面这条命令检查你的电脑上有哪些显卡,安装了什么驱动
lspci -k | grep -A 2 -i "VGA"
比如我的显示是这样的,
01:00.0 VGA compatible controller: NVIDIA Corporation Device 1b80 (rev a1)
Subsystem: Gigabyte Technology Co., Ltd Device 3702
Kernel driver in use: nouveau
1.2 然后查看哪一些驱动是系统推荐安装的,我们需要的是nvidia产家的驱动。
sudo ubuntu-drivers devices
比如我的是这样的,
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001B80sv00001458sd00003702bc03sc00i00
vendor : NVIDIA Corporation
driver : xserver-xorg-video-nouveau - distro free builtin
driver : nvidia-384 - distro non-free recommended
== cpu-microcode.py ==
driver : intel-microcode - distro non-free
从中可以看出,推荐安装的是nvidia-384版本的显卡驱动
1.3 禁用nouveau,如果系统没有该文件,则自己创建该文件
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在最后加入两行,
blacklist nouveau options nouveau modeset=0
命令行执行,
sudo update-initramfs -u
修改后需要重启系统。
1.4 确认nouveau是否已经被干掉,使用命令,
lsmod | grep nouveau
如果没有显示内容,则说明禁用成功。
1.5 安装GPU驱动
添加graphics driver ppa
sudo add-apt-repository ppa:graphics-drivers/ppa
第一次运行,可能会出现警告,回车后继续执行下列命令
sudo apt-get update sudo apt-get install nvidia-384 apt install mesa-common-dev apt install freeglut3-dev
重启系统
查看GPU相关信息
nvidia-smi
出现以上信息说明安装成功。