From f2fb294dcfbc97bc1a64c7a998cf87567fb5585c Mon Sep 17 00:00:00 2001 From: Huanyu He Date: Thu, 20 Feb 2025 15:00:12 -0800 Subject: [PATCH] adding tensordict dependency in validate_binaries.sh (#2756) Summary: # context * failed to directly install torchrec from nightly build: P1736807637 * the reason is that tensordict is not available from https://download.pytorch.org/whl/nightly/* * so the tensordict has to be installed separately # changes * install current tensordict version in the validate_binaries.sh script Differential Revision: D69950708 --- .github/scripts/validate_binaries.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index f9958a208..076672bbb 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -80,6 +80,9 @@ conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL" # install requirements from pypi conda run -n "${CONDA_ENV}" pip install torchmetrics==1.0.3 +# install tensordict from pypi +conda run -n "${CONDA_ENV}" pip install tensordict==0.7.1 + # install torchrec conda run -n "${CONDA_ENV}" pip install torchrec --index-url "$PYTORCH_URL"