get_gpu_device_count
Return the number of available GPU devices.
Note
- The returned value is
0
if the installed or compiled package does not support training on GPU. - Use the
CUDA_VISIBLE_DEVICESenvironment variable to limit the list of available devices.
Method call format
get_gpu_device_count()
Usage examples
from catboost.utils import get_gpu_device_count
print('I see %i GPU devices' % get_gpu_device_count())