torch.cuda.device_count#
- torch.cuda.device_count()[source]#
Return the number of GPUs available.
Note
This API will NOT poison fork if NVML discovery succeeds. See Poison fork in multiprocessing for more details.
- Return type
To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.
