TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

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.

Ask DeepWiki