device

dragon.vm.dali.device(
  device_type,
  device_index=0
)[source]

Context-manager to nest the the device spec.

Examples:

with dali.device('cuda', 0):
    pass
Parameters:
  • device_type ({'cpu', 'gpu', 'cuda'}, required) – The type of device.
  • device_index (int, optional, default=0) – The index of the device.