tensorplay.cuda.device_count

tensorplay.cuda.device_count() → int[source]

Return the number of GPUs available.

Note

This API will NOT poison fork if NVML discovery succeeds.