Table Of Contents
Table Of Contents

num_gpus

mxnet.context.num_gpus()[source]

Query CUDA for the number of GPUs present.

Raises:Will raise an exception on any CUDA error.
Returns:count – The number of GPUs.
Return type:int