How to use GPU only on the nodes that have it

Hi,

I have a cluster with 8 nodes, with 4 cores each.

Only node 1 has a GPU. If I launch the command

spmd
gpuDeviceCount("available")
end

I get that the first four labs have answer "1", and the other labs have answer "0. So everything as expected. In the code I want to run, some operations will be performed on the GPU, some others not. I am testing and I tried to create a gpuArray, but I get the following error

>> spmd
gpuArray([2 2 2]);
end
Error detected on workers 9 10 11 12 13 14 15 21 22 23 24 25 26 27 28 29 30 31 32.

Caused by:
    Failed to load graphics driver. Unable to load library 'libcuda.so.1'. The error was:
    libcuda.so.1: cannot open shared object file: No such file or directory
    Update or reinstall your graphics driver. For more information on GPU support, see GPU Support by Release.

How can I tell Matlab that the array shall be created only in the node that has the GPU, namely, node 1 ?

Best

Maria

[Ive J]

what about this?

spmd
    if gpuDeviceCount
        gpuArray([2 2 2])
    else
        % do something else
    end
end

Answer 1

[Joss Knight]

canUseGPU is the favoured way to guard your code.