Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[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

Open
hpx502766238 opened this issue Jul 10, 2023 · 1 comment
Labels
ep:DML issues related to the DirectML execution provider feature request request for unsupported feature or enhancement

Comments

@hpx502766238
Copy link

hpx502766238 commented Jul 10, 2023

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}])

@hpx502766238 hpx502766238 added the feature request request for unsupported feature or enhancement label Jul 10, 2023
@github-actions github-actions bot added the ep:DML issues related to the DirectML execution provider label Jul 10, 2023
@hpx502766238
Copy link
Author

"So, when using ONNX Runtime CUDA, it is equally important to understand the mapping between the ID and the corresponding GPU."

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ep:DML issues related to the DirectML execution provider feature request request for unsupported feature or enhancement
Projects
None yet
Development

No branches or pull requests

1 participant