Function at::cuda::is_available

• Defined in File CUDAContext.h

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.)