How to use multiple GPUs in distributed computing server?

2 次查看(过去 30 天)
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
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

类别

Help CenterFile Exchange 中查找有关 Parallel Computing 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by