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_DEVICES environment 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()