[Feature Request] Python DirectML,I can set 'device_id=1' to choose a GPU,but I can't know which ID corresponds to which GPU. #16644
Labels
ep:DML
issues related to the DirectML execution provider
feature request
request for unsupported feature or enhancement
Describe the feature request
[Feature Request] I use Python with ONNX Runtime DirectML. I can set 'device_id=1' to choose a GPU, but I can't determine which GPU corresponds to which ID. Can you add an API to list all the devices with their IDs that support DirectML?
Describe scenario use case
python onnxruntime-directml,I use it to do some audio AI inference,there is a example:
print(onnxruntime.get_available_providers())
model_path = "model.onnx"
session = onnxruntime.InferenceSession("model.onnx",providers=['DmlExecutionProvider'])
session.set_providers(['DmlExecutionProvider'], [ {'device_id': 0}])
The text was updated successfully, but these errors were encountered: