Rate this Page
★★★★★

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 variablePYTORCH_NVML_BASED_CUDA_CHECK=1 is set. For more details, seePoison fork in multiprocessing.

Return type

bool