draugr.torch_utilities.system.device.select_cuda_device

draugr.torch_utilities.system.device.select_cuda_device(cuda_device_idx: int) device[source]
Parameters

cuda_device_idx

Returns

Return type