【发布时间】:2016-09-13 15:44:54
【问题描述】:
最近我尝试在大型 CPU 服务器上实施和部署深度学习解决方案(多层 LSTM 网络,带有用于静态输入的附加层)。经过多次尝试,与我的个人计算机上的性能相比,我的速度仅提高了 3 倍。我听说 GPU 可能会做得更好。你能解释一下为什么在深度神经网络计算方面 GPU 比 CPU 好得多的确切原因是什么?
【问题讨论】:
-
深度学习(和 AI)严重依赖矩阵和矩阵运算(即线性代数)。矩阵运算可以高度并行化,并且可以利用数百个线程(例如两个矩阵的乘法)——这就是 GPU 非常擅长的(也可以考虑 GPU 的内存带宽)
标签: performance neural-network gpu cpu