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

[Documentation]: add MI300 series support on the dockerhub Overview #129

Open
seungrokj opened this issue May 27, 2024 · 2 comments
Open

Comments

@seungrokj
Copy link

Description of errors

https://hub.docker.com/r/rocm/pytorch

Can you please add this ? Many are confused as the Overview does not contain mi300 series information.

Starting from ROCm6.0, the "latest" tag docker images also contain MI300 (gfx942) support.

Attach any links, screenshots, or additional evidence you think will be helpful.

No response

@yiakwy-xpu-ml-framework-team

Hi @seungrokj as other friends suggested, MI300 are supported from rocm6.0, "but the dockerhub view does not reflect this change". So I suggested you

docker pull rocm/pytorch:latest-release

or

docker pull rocm/pytorch:rocm6.1.1_ubuntu22.04_py3.10_pytorch_release-2.1.2

Since I know the lastest NV-pytorch is better supported via pytorch 2.1+ and python 3.10.

@seungrokj
Copy link
Author

seungrokj commented May 27, 2024

In this screen shot.. it's better to put mi300 information here, otherwise people will keep asking whether the rocm/pytorch:latest-release supports gfx942 or not.

image

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