在具有多个GPU的计算机上运行CNTK

时间:2017-02-15 14:04:50

标签: cntk

我在具有2个K80 GPU的Azure VM上运行CNTK。

我是否需要在Python(Jupyter Notebook)代码中执行任何特殊操作以利用多个GPU或CNTK是否会自动利用这两个GPU?

1 个答案:

答案 0 :(得分:2)

目前,通过MPI支持CNTK的自动多GPU培训。使用multiprocessing可以在Notebook中启动多GPU培训。

Here是有关如何使用多个GPU运行CNTK的更多详细信息。