From 7c266e6a0132dab4be81a0eb5f955258822a48ec Mon Sep 17 00:00:00 2001 From: henrylhtsang <91030427+henrylhtsang@users.noreply.github.com> Date: Tue, 30 Jan 2024 11:14:02 -0800 Subject: [PATCH] Update validate_binaries.sh --- .github/scripts/validate_binaries.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 591900ee0..3fbb19e42 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -12,7 +12,7 @@ conda create -y -n build_binary python="${MATRIX_PYTHON_VERSION}" conda run -n build_binary python --version -conda run -n build_binary python -m pip cache purge +conda run -n build_binary pip cache purge # Install pytorch, torchrec and fbgemm as per # installation instructions on following page @@ -63,7 +63,7 @@ else conda run -n build_binary pip install torch --index-url "$PYTORCH_URL" # install fbgemm - conda run -n build_binary pip install fbgemm-gpu --index-url "$PYTORCH_URL" + conda run -n build_binary pip install --no-cache-dir fbgemm-gpu --extra-index-url "$PYTORCH_URL" # install requirements from pypi conda run -n build_binary pip install torchmetrics==1.0.3