ray.train.torch.get_device#

ray.train.torch.get_device() → torch.device[source]#

Gets the correct torch device to use for training.

PublicAPI (beta): This API is in beta and may change before becoming stable.