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

DeepSpeed on ROCm 6 #825

Open
MisterNorwood opened this issue Oct 15, 2024 · 1 comment
Open

DeepSpeed on ROCm 6 #825

MisterNorwood opened this issue Oct 15, 2024 · 1 comment

Comments

@MisterNorwood
Copy link

While the project does not support AMD natively, I have seen people run the model with relative ease.
Looking through documentation and repositories, I think it is possible to make DeepSpeed work in rocm, since it is officially supported. My question is if anyone has gotten it to work and if, how can one do so? Default script runs installation for CUDA and expects CUDA libraries.

@rose07
Copy link

rose07 commented Oct 16, 2024

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

No branches or pull requests

2 participants