diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index cb6d09f5bb..28894e368a 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -1,4 +1,4 @@ -ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.2-complete +ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.4-complete ARG TRITON_BRANCH="ba5c1517" ARG TRITON_REPO="https://github.com/ROCm/triton.git" ARG PYTORCH_BRANCH="8514f051" # release/2.10 as of 3/17