深度学习环境搭建——安装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

出现以上信息说明安装成功。

转载于:https://www.cnblogs.com/tinrry/p/9614668.html