Nerogar / OneTrainer

OneTrainer is a one-stop solution for all your stable diffusion training needs.
GNU Affero General Public License v3.0
1.73k stars 145 forks source link

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

Open Reaper176 opened 1 week ago

Reaper176 commented 1 week ago

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 commented 1 week ago

.

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

gambikules commented 1 week ago

same here

Nerogar commented 1 week ago

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 commented 1 week ago

The correct package for ROCm is onnxruntime-rocm.

Nerogar commented 2 days ago

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 commented 2 days ago

https://github.com/microsoft/onnxruntime/issues/18052 Looks like they are not going to upload onnxruntime-rocm builds to pypi :/