Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

AMD ROCm installer #2167

Merged
merged 2 commits into from
Mar 27, 2024
Merged

AMD ROCm installer #2167

merged 2 commits into from
Mar 27, 2024

Conversation

Disty0
Copy link
Contributor

@Disty0 Disty0 commented Mar 27, 2024 •
edited
Loading

  • Add ROCm support to setup.sh and gui.sh with --use-rocm argument.
  • Separate onnxruntime from requirements.txt and make it platform specific (to support Add OpenVINO and ROCm ONNX Runtime for WD14 kohya-ss/sd-scripts#1213).
    • ROCm: onnxruntime-training with ROCm support
    • IPEX: onnxruntime-openvino
    • CUDA: onnxruntime-gpu (no change, untested)
    • MacOS: onnxruntime (untested)

I went with onnxruntime for MacOS since it has no CUDA. Not sure if there is any better option for MacOS.
I went with PyTorch ROCm Nightly builds for ROCm 6.0 support.

@bmaltais
Copy link
Owner

Thank you for your contribution!

@bmaltais bmaltais merged commit 5d4c448 into bmaltais:dev Mar 27, 2024
1 check failed
@Disty0 Disty0 mentioned this pull request Apr 9, 2024
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Disty0 @bmaltais