Search code examples
cntk

Run CNTK on machine with multiple GPUs


I'm running CNTK on an Azure VM with 2 x K80 GPUs.

Do I need to do anything special in my Python (Jupyter Notebook) code in order to take advantage of the multiple GPUs or will CNTK automatically take advantage of both GPUs?


Solution

  • Currently automatic multi-GPU training in CNTK is supported via MPI. Launching multi-GPU training in Notebook is possible using multiprocessing.

    Here is more details on how to run CNTK with multiple GPUs.