can't install onnxruntime-gpu for cuda12.x

astral-sh / rye

a Hassle-Free Python Experience

https://rye.astral.sh

MIT License

13.46k stars 460 forks source link

can't install onnxruntime-gpu for cuda12.x #976

Open JaheimLee opened 5 months ago

JaheimLee commented 5 months ago

The defalt cuda version for onnxruntim-gpu from pypi is cuda11.8. But I want to install the cuda12.x version. So I add extra url in myproject.toml like

[[tool.rye.sources]]
name = "onnxruntime-gpu"
url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"

However, rye add onnxruntime-gpu can only install the default version.

iKora128 commented 1 month ago

I have a same issue... Did you solve this problem???