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