PNY NVIDIA Quadro RTX 4000 是用于 Linux 机器学习的优秀 GPU 吗?

数据挖掘 显卡 linux
2022-02-20 17:26:28

作为一名网络开发人员,我对数据科学/机器学习越来越感兴趣,以至于我决定在家里建立一个实验室。

我发现了 Quadro RTX 4000,我想知道它在 Ubuntu Linux 上运行机器学习框架的效果如何。Linux 上是否提供了正确的驱动程序,以便这张卡可以利用 ML 框架?

LINUX X64 (AMD64/EM64T) 显示驱动

这是我能找到的唯一驱动程序,但它是一个“显示驱动程序”,所以我不确定这是否使 ML 框架能够使用此 GPU 进行加速。它适用于基于英特尔的处理器吗?

任何指导将不胜感激。

1个回答

您似乎正在查看最新的 Quatro 4000,它具有以下计算等级:

在此处输入图像描述

您可以在此处找到所有 Nvidia GPU的完整列表。

虽然它似乎有一个令人印象深刻的 7.5 分(与 RTX 20180ti 相同),但主要减少了 8Gb 的内存。这绝对足以开始使用 ML/DL,并且可以让您做很多事情。但是,内存通常会减慢您的速度并限制您的模型。

原因是大型模型将需要大量参数。看看下表(Keras 中包含的模型),您可以在其中看到每个模型所需的参数数量:

在此处输入图像描述

问题是您拥有的参数越多,您需要的内存就越多,因此您在训练期间能够使用的批大小越小。有很多关于更大与更小批量大小的论据 - 但是在使用大型模型时,内存较少将迫使您仍然使用更小的批量大小。

从 Nvidia 的营销看来,Quadro 产品线更针对创意开发人员(电影/图像编辑等),而 Geforce 系列则用于游戏 AI。这凸显了 Quadro 不一定针对快速计算进行优化。