[Bug]: ERROR: No matching distribution found for onnxruntime-gpu==1.18.0

[Reaper176]

What happened?

installation fails

What did you expect would happen?

.

Relevant log output

Unpacking resources
Unpacking resources
Download Complete
Looking in indexes: https://download.pytorch.org/whl/rocm5.7
Collecting torch==2.3.1+rocm5.7 (from -r requirements-rocm.txt (line 3))
  Using cached https://download.pytorch.org/whl/rocm5.7/torch-2.3.1%2Brocm5.7-cp310-cp310-linux_x86_64.whl (1905.1 MB)
Collecting torchvision==0.18.1+rocm5.7 (from -r requirements-rocm.txt (line 4))
  Using cached https://download.pytorch.org/whl/rocm5.7/torchvision-0.18.1%2Brocm5.7-cp310-cp310-linux_x86_64.whl (65.5 MB)
ERROR: Could not find a version that satisfies the requirement onnxruntime-gpu==1.18.0 (from versions: none)
ERROR: No matching distribution found for onnxruntime-gpu==1.18.0
Could not install OneTrainer (StabilityMatrix.Core.Exceptions.ProcessException: pip install failed with code 1: 'Looking in indexes: https://download.pytorch.org/whl/rocm5.7\nCollecting torch==2.3.1+rocm5.7 (from -r requirements-rocm.txt (line 3))\n  Using cached https://download.pytorch.org/whl/rocm5.7/torch-2.3.1%2Brocm5.7-cp310-cp310-linux_x86_64.whl (1905.1 MB)\nCollecting torchvision==0.18.1+rocm5.7 (from -r requirements-rocm.txt (line 4))\n  Using cached https://download.pytorch.org/whl/rocm5.7/torchvision-0.18.1%2Brocm5.7-cp310-cp310-linux_x86_64.whl (65.5 MB)\nERROR: Could not find a version that satisfies the requirement onnxruntime-gpu==1.18.0 (from versions: none)\nERROR: No matching distribution found for onnxruntime-gpu==1.18.0\n'

Output of pip freeze

.

[O-J1]

.

This cannot be troubleshooted unless you include your output of pip freeze. Additionally what OS are you running this on?

[gambikules]

same here

[Nerogar]

The actual error message is

ERROR: Could not find a version that satisfies the requirement onnxruntime-gpu==1.18.0 (from versions: none)
ERROR: No matching distribution found for onnxruntime-gpu==1.18.0

[hameerabbasi]

The correct package for ROCm is onnxruntime-rocm.

[Nerogar]

The correct package for ROCm is onnxruntime-rocm.

Where did you find that package? At least on pypi I only see onnxruntime and onnxruntime-gpu

[Heasterian]

microsoft/onnxruntime#18052 Looks like they are not going to upload onnxruntime-rocm builds to pypi :/

[rodrigoandrigo]

The correct package for ROCm is onnxruntime-rocm.

Where did you find that package? At least on pypi I only see onnxruntime and onnxruntime-gpu

has only onnxruntime_rocm https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/