Skip to content

Conversation

charlifu
Copy link

@charlifu charlifu commented Apr 17, 2025

This PR updates base image to rocm 6.4, newer hipblaslt, triton 3.3, torch 2.7, torch vision 0.22.

And hipblas-common is part of rocm-libs in rocm 6.4, so we do not need to install it manually in the docker file.

Signed-off-by: charlifu <[email protected]>
Copy link
Collaborator

@shajrawi shajrawi left a comment

Choose a reason for hiding this comment

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

Great work Charlie!

@charlifu charlifu merged commit 1c0a1ae into main Apr 17, 2025
4 of 6 checks passed
@gshtras gshtras deleted the charlifu/update_base_image branch September 25, 2025 14: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