深度学习环境搭建——安装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 然后查看哪一个驱动是系统推荐安装的

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-375版本的显卡驱动

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

猜你喜欢

转载自www.cnblogs.com/tinrry/p/9614668.html