Add logging for torch nightly version (#17669)

Signed-off-by: Yang Wang <elainewy@meta.com>
2025-05-06 17:45:51 -07:00
parent ed3a1d2106
commit 6de3e13413
2 changed files with 12 additions and 2 deletions
--- a/docker/Dockerfile.nightly_torch
+++ b/docker/Dockerfile.nightly_torch
@@ -309,5 +309,7 @@ ENV HF_HUB_ENABLE_HF_TRANSFER 1
 RUN --mount=type=cache,target=/root/.cache/uv \
    uv pip install --system -r requirements/nightly_torch_test.txt

-#################### UNITTEST IMAGE #############################
+# Logging to confirm the torch versions
+RUN pip freeze | grep -E 'torch|xformers|vllm|flashinfer'

+#################### UNITTEST IMAGE #############################