Device¶
- metatensor.models.utils.devices.pick_devices(architecture_devices: List[str], desired_device: str | None = None) List[device] [source]¶
Pick (best) devices for training.
The choice is made on the intersection of the
architecture_devices
and the available devices on the current system. If nodesired_device
is provided the first device of this intersection will be returned.- Parameters:
- Return type: