syne_tune.num_gpu module

Adapted from to not run in Shell mode which is unsecure. https://github.com/aws/sagemaker-rl-container/blob/master/src/vw-serving/src/vw_serving/sagemaker/gpu.py

syne_tune.num_gpu.get_num_gpus()

Returns the number of available GPUs based on configuration parameters and available hardware GPU devices. Gpus are detected by running “nvidia-smi –list-gpus” as a subprocess. :rtype: int :return: (int) number of GPUs