[Feature Request] Easier installation of onnxruntime-gpu
#20200
Labels
ep:CUDA
issues related to the CUDA execution provider
feature request
request for unsupported feature or enhancement
platform:windows
issues related to the Windows platform
Describe the feature request
I would like to see an easier installation of the pypi package
onnxruntime-gpu
. Ideally the package will automatically install:Where XX denotes the depending CUDA version. This should work for Linux and windows equally and heavily ease the entrypoint by not requiring putting all the needed libs on the PATH.
Ideally it would even be possible to install the needed libraries for cuda and/or art only by doing
pip install onnxruntime-gpu[cuda]
I am happy to contribute or help testing for this, but I could use someone giving me hints where to look.
@jywu-msft for viz
Describe scenario use case
This will positively impact all python users of ONNX Runtime that use GPU acceleration. Hopefully also reducing library installation issues on GitHub significantly.
The text was updated successfully, but these errors were encountered: