Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit f9d0a29

Browse files
authored
Update install_requirements.sh
1 parent 8cb415d commit f9d0a29

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

install/install_requirements.sh

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ echo "Using pip executable: $PIP_EXECUTABLE"
6565
PYTORCH_NIGHTLY_VERSION=dev20241118
6666

6767
# Nightly version for torchvision
68-
VISION_NIGHTLY_VERSION=dev20241118
68+
VISION_NIGHTLY_VERSION=dev20241114
6969

7070
# Nightly version for torchtune
7171
TUNE_NIGHTLY_VERSION=dev20241010
@@ -92,7 +92,6 @@ fi
9292
# pip packages needed by exir.
9393
REQUIREMENTS_TO_INSTALL=(
9494
torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
95-
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
9695
torchtune=="0.4.0.${TUNE_NIGHTLY_VERSION}"
9796
)
9897

@@ -104,6 +103,13 @@ REQUIREMENTS_TO_INSTALL=(
104103
"${REQUIREMENTS_TO_INSTALL[@]}"
105104
)
106105

106+
# Workaround until torchvision nightly is gets bumped
107+
(
108+
set -x
109+
$PIP_EXECUTABLE install --no-deps --extra-index-url "${TORCH_NIGHTLY_URL}" \
110+
torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
111+
)
112+
107113
(
108114
set -x
109115
$PIP_EXECUTABLE install torchao=="0.5.0"

0 commit comments

Comments
 (0)