ray.rllib.utils.torch_utils.get_device#

ray.rllib.utils.torch_utils.get_device(config)[source]#

Returns a torch device edepending on a config and current worker index.