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.

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

AMD + Linux #84

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

AMD + Linux #84

wants to merge 11 commits into from

Conversation

YourSandwich
Copy link

Hello, this is my first Merge request ever, so please be kind to me 🙏

I like this project and find it useful. I encountered some initial problems while setting this up on my AMD system, as mentioned in this issue: #55

Therefore, I've created a script for automatic deployments on Linux AMD systems.

There is only one issue left: the file src/mdx.py has the torch provider set to CUDAExecutionProvider. Changing it to ROCMExecutionProvider was enough to resolve the ONNX errors.

Since I am not experienced with Torch and ONNX, I was unable to create a function that would set the Provider automatically.

@ISAKEM
Copy link

ISAKEM commented Apr 1, 2024

Can confirm this works on Arch with a RX 6800 XT, no issues so far

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants