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

Update pytorch for ROCm & CUDA #185

Closed
wants to merge 2 commits into from
Closed

Update pytorch for ROCm & CUDA #185

wants to merge 2 commits into from

Conversation

Delaunay
Copy link
Collaborator

No description provided.

@Delaunay Delaunay changed the title Use rocm6.0 Update pytorch for ROCm & CUDA Jan 17, 2024
@Delaunay Delaunay marked this pull request as ready for review January 17, 2024 19:47
@@ -1 +1 @@
--extra-index-url https://download.pytorch.org/whl/cu118
--extra-index-url https://download.pytorch.org/whl/nightly/cu118

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we bump CUDA to the latest 12.1?

https://download.pytorch.org/whl/nightly/cu121

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

12.1 does not support the CI gpus

@Delaunay Delaunay closed this Jun 12, 2024
@Delaunay Delaunay deleted the mi300_patch branch June 12, 2024 13:51
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