Function at::cuda::is_available

Function Documentation

bool at::cuda::is_available()

CUDA is available if we compiled with CUDA, and there are one or more devices.

Returns true if at least one CUDA device is available.

If we compiled with CUDA but there is a driver problem, etc., this function will report CUDA is not available (rather than raise an error.)


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources