diff --git a/prebuilts/Dockerfile b/prebuilts/Dockerfile index f1d8f16..1540f1a 100644 --- a/prebuilts/Dockerfile +++ b/prebuilts/Dockerfile @@ -1,5 +1,5 @@ -ARG IMAGE_NAME=rocm/dev-ubuntu-22.04:${ROCM_VERSION} -FROM ${IMAGE_NAME} +ARG ROCM_VERSION=0.0 +FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION} ENV DEBIAN_FRONTEND=noninteractive \ PYTHONUNBUFFERED=1 \