diff --git a/docker/Dockerfile b/docker/Dockerfile index aa8c71896..46715a4e6 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -570,6 +570,10 @@ elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \ python3 -m pip install "cuda-python>=13,<14" ; \ fi +# Install the latest available AI Dynamo prerelease from NVIDIA's package index. +RUN --mount=type=cache,target=/root/.cache/pip \ + python3 -m pip install --pre --extra-index-url https://pypi.nvidia.com/ ai-dynamo + # Add yank script COPY --chown=root:root --chmod=755 docker/configs/yank /usr/local/bin/yank