From 7ddac90351adbcc1f992b9df6073a0e5ddd7ee14 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Wed, 13 Nov 2024 17:58:04 -0800
Subject: [PATCH] remove_pinned

---
 .github/scripts/setup-env.sh | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh
index 0d962d154f2..adb1256303f 100755
--- a/.github/scripts/setup-env.sh
+++ b/.github/scripts/setup-env.sh
@@ -36,7 +36,6 @@ conda activate ci
 conda install --quiet --yes libjpeg-turbo -c pytorch
 pip install --progress-bar=off --upgrade setuptools==72.1.0
 
-
 # See https://github.com/pytorch/vision/issues/6790
 if [[ "${PYTHON_VERSION}" != "3.11" ]]; then
   pip install --progress-bar=off av!=10.0.0
@@ -78,10 +77,7 @@ case $GPU_ARCH_TYPE in
     ;;
 esac
 PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}"
-#pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
-
-#install torch built with manylinux 2.28
-pip install --progress-bar=off --pre torch==2.6.0.dev20241106 --index-url="${PYTORCH_WHEEL_INDEX}" --force-reinstall
+pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
 
 if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
   python -c "import torch; exit(not torch.cuda.is_available())"