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

Question

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

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

Answer 1

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

Here 详细介绍了如何在多个 GPU 上运行 CNTK。