diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 6f8c7222fd..a702bb9f52 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -1,4 +1,4 @@ -ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete +ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2-complete ARG TRITON_BRANCH="57c693b6" ARG TRITON_REPO="https://github.com/ROCm/triton.git" ARG PYTORCH_BRANCH="89075173"