[Question]: hub.docker.com/r/rocm/pytorch

[rasemailcz]

Suggestion Description

Hi,

1. is this the right place to ask for some (specific) images (like ROCm 6.0.2 + python 3.11 + pytorch ?) at https://hub.docker.com/r/rocm/pytorch ?
2. if so, are the rocm/pytorch Dockerfiles available somewhere?

Operating System

No response

GPU

No response

ROCm Component

No response

[jitendra42]

The dockerfile seems to be in pytorch repo:
https://github.com/pytorch/pytorch/blob/main/.ci/docker/ubuntu-rocm/Dockerfile
https://github.com/pytorch/pytorch/blob/main/.ci/docker/centos-rocm/Dockerfile