TensorFlow 的公有云 GPU 支持

数据挖掘 机器学习 张量流 显卡
2021-09-15 17:39:03

我正在使用 TensorFlow 构建卷积神经网络。我有最新的 Mac Pro,使用了我所有的内核,但我仍然需要几个小时来训练一个网络。我需要对许多不同的调整参数进行网格搜索,所以这并不理想。

这是一个个人项目,所以我想通过使用云来省钱。我已经广泛使用 AWS,我喜欢使用他们的现货市场的想法,但是我听说 1) Amazon 的 GPU 实例与 TensorFlow 开箱即用不兼容(AWS 的 GPU 具有 CUDA 3.0,而 TensorFlow 需要 3.5)和2) 亚马逊的 GPU 有点慢。

使用 TensorFlow 时,微软的 Azure 或谷歌的 Google Compute Engine 是否有易于使用的 GPU?在相关的神经网络训练技术规格、价格和易用性方面,他们的 GPU 与亚马逊的 GPU 相比如何?您是否会推荐较小的云提供商?

此外,与顶级 Mac Pro 相比,我期望在其中一个虚拟 GPU 上进行训练的速度大概能快多少?

2个回答

微软几周前在 Azure 中宣布了带有GPU的虚拟机。他们使用 K-80 NVIDIA 卡。最大的机器有 4 个 GPU 和 224 GB 的内存。

很好玩深度学习:-)

谷歌目前没有任何公开的 GPU 或 TPU 产品,但如果你为一家大公司工作,它们可能会提供帮助。直接联系他们。

资料来源:谷歌代表,就在今天。