ray.rllib.utils.tf_utils.get_gpu_devices#

ray.rllib.utils.tf_utils.get_gpu_devices() → List[str][source]#

Returns a list of GPU device names, e.g. [“/gpu:0”, “/gpu:1”].

Supports both tf1.x and tf2.x.

Returns:: List of GPU device names (str).