教授，我的想法是，我们可以用GPU来做神经网络的计算！”亚历克斯说出了自己的计划。
伊利亚在旁边听得直皱眉。
“GPU？那是用来渲染三维游戏多边形的硬件。我们做的是深度神经网络训练，底层逻辑不兼容吧？”伊利亚提出质疑。
亚历克斯立刻转身面对伊利亚，开始在白板上列数据。
“表面上看是两码事，但实际上完美契合。你想想，我们现在的代码用CPU跑，为什么这么慢？”亚历克斯反问。
伊利亚思索了一下。
“因为要处理庞大的矩阵乘法运算，数据量太大了。”伊利亚如实回答。
亚历克斯重重地敲了敲白板。
“这就对了！CPU的核心数量很少，一般只有两到八个。它的架构设计是为了处理操作系统里复杂的逻辑分支。”亚历克斯开始分析。
接着，他在旁边写下英伟达显卡的参数。
“但你看GPU。它精简了复杂的控制逻辑，把芯片面积全留给了几百上千个小计算核心。”亚历克斯指着那些数字。
“深度神经网络的核心运算是什么？就是海量的加减乘除和取最大值。根本不需要复杂的逻辑跳转。”亚历克斯看着两人。
“让几个全能的CPU核心排队去处理上百万次简单的乘法运算，这完全是资源错配。”亚历克斯抛出核心论点。
“而GPU的几百个核心，可以把这上百万次乘法拆解开，同步进行并行计算。这才是真正的算力匹配！”亚历克斯提高了音量。
教研室里安静了几秒钟。
伊利亚的大脑飞速运转，推演着亚历克斯的设想。
“如果真能把我们的代码通过CUDA接口迁移到GPU上……”伊利亚喃喃自语。
“几百个核心同时进行矩阵运算，训练速度绝对会成百倍上升！以前需要一个月的模型，可能两天就能跑出结果！”亚历克斯接上了伊利亚的话。
本章未完，请点击"下一页"继续阅读！ 第2页 / 共4页