TensorFlow 会利用 GPU 进行预测分析吗？

will TensorFlow utilize GPU for predictive Analysis?

GPU 适合并行计算，但问题是一些机器学习库不使用 GPU，除非基于图像处理或某种图形处理的机器学习，如果我使用机器学习进行预测怎么办分析？像 TensorFlow 这样的库会使用 GPU 吗？或者他们只使用 CPU？或者我可以选择使用哪个处理单元？这是怎么回事？

注意：预测分析不需要图形处理。

简短的回答：是的，会的！

稍长的答案：

在任何支持 GPU 的机器学习框架中，GPU 中发生的计算不限于图形处理。例如，如果您的模型是一个简单的逻辑回归，那么 TensorFlow 等框架如果配置正确，将 运行 在 GPU 上运行。

GPU 用于机器学习的优势在于，训练大型神经网络可以从 GPU 提供的高度并行性中获益匪浅。

如果你想了解更多，我建议你开始here or here。

需要考虑的一些事项：

  • 模型从 GPU 中 运行 中受益的程度取决于它从一般并行计算中受益的程度。
  • 深度学习模型可以应用于预测分析，以及更经典的机器学习模型。请记住，神经网络可能是从 GPU 中受益的模型类别（参见上面的链接）。
  • 尽管 运行使用 GPU（或什至更专业的硬件）的宁模型可以带来好处，但我建议您不要选择框架，尤其是不要选择仅基于算法的算法事实上，它将受益于并行性，而是查看 给定算法对您拥有的数据有多合适。