[Docker] Install AI Dynamo nightly (#36946)

2026-08-28 23:27:52 -07:00
parent d1ce017665
commit 4d7d2ebb44
1 changed files with 4 additions and 0 deletions
@@ -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