torch.cuda.is_available

torch.cuda.is_available() [source]

Return a bool indicating if CUDA is currently available.

Note

This function will NOT poison fork if the environment variable PYTORCH_NVML_BASED_CUDA_CHECK=1 is set. For more details, see Poison fork in multiprocessing.

Return type

bool