mxnet.npx.num_gpus

num_gpus()

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

---

Did this page help you?

Yes

No

Thanks for your feedback!

---