0

我想使用Cloud ML上的多個GPU運行Tensorflow應用程序。Google Cloud ML是否僅支持分佈式Tensorflow以用於多個GPU培訓工作?

我Tensorflow應用程序是用在非分佈式模式,即概括here

據我瞭解,如果我想使用雲ML運行此相同的應用程序,具有多重GPU的,那麼應用程序必須使用規模CUSTOM層,我需要設置參數服務器,工作服務器,這似乎是一個分佈式張量流模式。 Link here

這是在Cloud ML上運行多個GPU培訓作業的唯一方法嗎?

是否有幫助我將我的multiGPU(基於塔)培訓應用程序所需的更改範圍擴展到分佈式tensorflow應用程序的指南?

回答

1

您可以將CUSTOM層僅使用一個主節點,而不使用工作站/參數服務器。這些是可選參數。

Then complex_model_m_gpu has 4 GPUs and complex_model_l_gpu has 8.

相關問題