2017-02-15 35 views
3

我在具有2個K80 GPU的Azure VM上運行CNTK。在具有多個GPU的機器上運行CNTK

我是否需要在我的Python(Jupyter Notebook)代碼中做任何特殊的事情來利用多個GPU,或者CNTK會自動利用這兩種GPU?

回答

2

目前通過MPI支持CNTK中的自動多GPU訓練。使用multiprocessing可以啓動Notebook中的多GPU培訓。

Here是關於如何使用多個GPU運行CNTK的更多細節。