Table Of Contents
Table Of Contents

mxnet.context.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