Skip to content

Commit 455d976

Browse files
TroyGardenfacebook-github-bot
authored andcommitted
adding tensordict dependency in validate_binaries.sh (#2756)
Summary: Pull Request resolved: #2756 # 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 Reviewed By: dstaay-fb Differential Revision: D69950708 fbshipit-source-id: 2afc83eccc8169f0bd80d3d117059f914d3e16b2
1 parent 7420484 commit 455d976

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@ conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL"
8080
# install requirements from pypi
8181
conda run -n "${CONDA_ENV}" pip install torchmetrics==1.0.3
8282

83+
# install tensordict from pypi
84+
conda run -n "${CONDA_ENV}" pip install tensordict==0.7.1
85+
8386
# install torchrec
8487
conda run -n "${CONDA_ENV}" pip install torchrec --index-url "$PYTORCH_URL"
8588

0 commit comments

Comments
 (0)