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

Make AMD supported #17

Open
dimalmfao opened this issue Jan 9, 2024 · 4 comments
Open

Make AMD supported #17

dimalmfao opened this issue Jan 9, 2024 · 4 comments
Labels
enhancement New feature or request help wanted Extra attention is needed

Comments

@dimalmfao
Copy link
Contributor

Make AMD cards supported. I'll maybe write the dockerfile, I'll see if I have time.
Just reminder to make AMD
because i think people have short memory.
replace your memory for a new-one too.

@Vectorrent Vectorrent added enhancement New feature or request help wanted Extra attention is needed labels Jan 10, 2024
@Vectorrent
Copy link
Contributor

Thanks for submitting this. AMD support should be doable, with a little bit of work.

@Vectorrent
Copy link
Contributor

I've taken this as far as I can take it, without an AMD GPU to develop with. I need someone to set...

DEVICE='amd'

...in their .env file, then to use the "build" task from controller.sh or controller.ps1.

@Vectorrent Vectorrent moved this to In progress in Development Jan 25, 2024
@dimalmfao
Copy link
Contributor Author

Sadly, I don't have control over an AMD GPU anymore...

@Vectorrent
Copy link
Contributor

I looked into this. While I think it should be possible, I have some questions:

  1. How do we expose an AMD GPU to Docker Compose?
  2. Can AMD make use of torch.cuda() calls? (I'm almost certain it can)
  3. Can AMD GPUs work on Windows (almost certainly not without WSL; but neither can Nvidia, I don't think)

I think the code base is mostly ready for trying to do this. Would just need an interested person with some technical experience, able to help finalize the process.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
Status: In progress
Development

No branches or pull requests

2 participants