draugr.torch_utilities.system.device.auto_select_available_cuda_device

draugr.torch_utilities.system.device.auto_select_available_cuda_device(expected_memory_usage_mb: int = 1024) device[source]

Auto selects the device with highest compute capability and with the requested memory available

Parameters

expected_memory_usage_mb

Returns

Return type