Skip to content

Device selection through onnxruntime-directml  #18716

@ramkrishna2910

Description

@ramkrishna2910

Describe the issue

Hi,

I am using directml through onnxruntime and my platform has multiple directx 12 supported devices. Is there a recommended way to enumerate and select specific devices through python APIs?

Thanks

To reproduce

ort.InferenceSession API does not provide an option for device selection.

onnx_session = ort.InferenceSession(onnx_file_path, sess_options, providers=['DmlExecutionProvider'])

Urgency

No response

Platform

Windows

OS Version

10

ONNX Runtime Installation

Released Package

ONNX Runtime Version or Commit ID

1.15.1

ONNX Runtime API

Python

Architecture

X64

Execution Provider

DirectML

Execution Provider Library Version

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    ep:DMLissues related to the DirectML execution providerplatform:windowsissues related to the Windows platformstaleissues that have not been addressed in a while; categorized by a bot

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions