You know, if this would have been open source, then a quick PR would have fixed that...
File "torch_directml\device.py", line 38, in device
raise Exception(f"Invalid device_id argument supplied {device_id}. device_id must be in range [0, {num_devices}).")
So the fix seems to be something like:
- raise Exception(f"Invalid device_id argument supplied {device_id}. device_id must be in range [0, {num_devices}).")
+ raise Exception(f"Invalid device_id argument supplied {device_id}. device_id must be in range [0, {num_devices - 1}].")
--> ] instead of ) and the number of devices is one-based, while the device-id is zero based.