How to use multiple GPUs in distributed computing server?

Hi, I'm using the MATLAB distributed computing server. I have 3 computers in a cluster, and each is quipped with the Nvidia Quadro M4000 GPU and is set to TCC mode. I want to use some or all the GPUs to do the computing. Since the GPU node cannot be displayed in the Admin Center, how could I decide which GPUs I am using in the computation? Thanks.

回答 (1 件)

Joss Knight
Joss Knight 2016 年 3 月 29 日

1 投票

Try gpuDeviceCount to determine how many GPUs a worker has access to, and then some manipulation of labindex to assign devices to workers. By default your cluster will have, in effect, run
spmd
gpuDevice(mod(labindex, gpuDeviceCount)+1);
end

カテゴリ

質問済み:

2016 年 3 月 24 日

回答済み:

2016 年 3 月 29 日

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by