ROCm-docker icon indicating copy to clipboard operation
ROCm-docker copied to clipboard
verified publisher icon ROCm

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

Open seungrokj opened this issue 1 year ago • 2 comments

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

seungrokj avatar May 27 '24 05:05 seungrokj

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.

yiakwy-xpu-ml-framework-team avatar May 27 '24 06:05 yiakwy-xpu-ml-framework-team

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

seungrokj avatar May 27 '24 06:05 seungrokj