驱动是指驱动计算机里软件的程序,驱动程序是硬件厂商根据操作系统编写的配置文件。没有驱动程序,计算机中的硬件就无法工作。如果没有NVIDIA驱动,我们无法使用英伟达的GPU进行深度学习计算。
我们从一台刚刚重装好的电脑开始。
首先我们先下载英伟达驱动。我的电脑是小米游戏本,GPU是GTX 1060。驱动选择如下:
下载好驱动后,在对应文件夹打开终端,运行如下代码:
sudo chmod +x NVIDIA-Linux-x86_64-440.64.run
sudo ./NVIDIA-Linux-x86_64-440.64.run
我们会看到很多个报错信息(因此上面这段代码会经常使用)。我们将通过解决一个一个错误的方式来讲解安装步骤。
第一个错误:Nouveau
Nouveau是由第三方为NVIDIA显卡开发的一个开源3D驱动,也没能得到NVIDIA的认可与支持。虽然Nouveau无法和NVIDIA官方私有驱动相提并论