tensorplay.cuda.is_available

tensorplay.cuda.is_available() → bool[source]

Return a bool indicating if CUDA is currently available.

Note

This function will NOT poison fork if the environment variable TENSORPLAY_NVML_BASED_CUDA_CHECK=1 is set.