随着深度学习的持续火热,Ubuntu系统得使用变得越来越广泛。为了配合深度学习的需求,现在Ubuntu系统也在持续提升它的方便性。通常,深度学习中需要使用GPU资源进行加速,但安装和配置GPU计算环境往往比较复杂。幸运的是,在最新的 18.04版本的Ubuntu系统,环境的配置已经变得越来越简单。下面就介绍在其中安装NVIDIA的显卡驱动的过程。
默认情况下,进入Ubuntu系统后,系统默认是使用主板上的集成显卡以及与之配套的默认显卡驱动,如下图所示。因此,我们需要做的事就是为自己的独立显卡(下面实验中的机器上配套有两块GeForce 1080Ti显卡)安装配套驱动。
首先,检测你的NVIDIA图形卡和推荐的驱动程序的模型。执行命令:
$ ubuntu-drivers devices
执行结果如下图中的左图所示。可以系统为GeForce 1080 Ti显卡推荐的驱动是nvidia-driver-390。或者你也可以从【Software&Updates】提供的图形界面中查看到类似的结果,如下面的右图所示。