从文生文到文生图,再到文生视频……如今,生成式AI已经成为提高企业生产效率,加速业务创新的核心技术,而这项技术的背后,离不开CPU、GPU、NPU等计算芯片的强力支撑。
在《解读AI通用计算芯片:GPU训练CPU推理,用最优的成本降低AI算力支出》这篇文章中,笔者简单介绍了CPU与GPU芯片的不同之处。除了这两种通用芯片之外,NPU与AI计算过程中扮演着重要的角色。那么,作为同样拥有大量计算单元的并行计算芯片,GPU与NPU又有何不同之处呢?本文将简单介绍。
什么是 GPU?
GPU(Graphics Processing
Unit,图形处理器)是一种专门设计用于处理图形和图像计算任务的计算硬件,用于加速图像/视频处理和渲染等图形任务,例如在电脑游戏中,就需要高性能的GPU进行实时的图像渲染。
最初,GPU是为提高计算机图形的渲染速度而开发的,但随着技术的发展,它们已经扩展到更广泛的计算领域,尤其是在需要大规模并行处理能力的场景中。例如,在生成式AI计算过程中,GPU常用于训练大型神经网络。目前很多科技公司使用 Nvidia 的企业级 H-100 GPU 来训练他们的大型语言模型 (LLM)。
什么是NPU?
NPU是一种用于加速神经网络操作的专用处理器,具有并行处理能力(能够同时执行多个操作),并且通过硬件架构优化,它们可以有效地执行推理和训练等 AI 和 ML 任务,包括AI和机器学习(ML)计算任务,例如面部识别,甚至训练 AI 系统等。
通过针对不同任务需求的特定硬件优化,NPU拥有更高的性能,同时由于能耗较低,能够实现更高的能效比,这也是NPU相较于CPU、GPU等通用处理器最大的优势之一。
GPU的主要特点及其优势
1)强大的并行计算性能。GPU拥有成千上万个浮点运算单元,这些计算核心可以同时处理多个计算任务,这使得GPU在处理并行计算任务时比传统的CPU(中央处理器)更加高效,更加适合进行大规模并行计算。例如,在深度学习和其他机器学习算法中,GPU发挥着关键作用,因为这些算法需要处理大量的矩阵运算,GPU的并行处理能力可以显著加速这些运算。
2)通用性。作为通用计算处理器,GPU不仅能够用于图形渲染,而且它所拥有的并行处理能力也更适用于广泛的计算密集型任务。这也是目前AI训练过程中广泛采用GPU的根本原因之一。
除此之外,通过技术如CUDA(Compute
Unified Device Architecture,由NVIDIA开发)和OpenCL(Open Computing Language),GPU可以用于执行通用计算任务,即所谓的GPGPU(General-Purpose computing on Graphics Processing Units)。