From 4d7d2ebb44447c672c795ff8092e24bc6225f99f Mon Sep 17 00:00:00 2001
From: ishandhanani <82981111+ishandhanani@users.noreply.github.com>
Date: Sat, 29 Aug 2026 01:27:52 -0500
Subject: [PATCH] [Docker] Install AI Dynamo nightly (#36946)

---
 docker/Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

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