Skip to content

Commit 4c7183c

Browse files
authored
Update torch/tune/vision pins to 1/19/25 (#1467)
* Update install_requirements.sh * Update pytorch minor version * Update install_requirements.sh
1 parent 2b555b1 commit 4c7183c

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

install/install_requirements.sh

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,13 @@ echo "Using pip executable: $PIP_EXECUTABLE"
5151
# NOTE: If a newly-fetched version of the executorch repo changes the value of
5252
# PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
5353
# package versions.
54-
PYTORCH_NIGHTLY_VERSION=dev20241218
54+
PYTORCH_NIGHTLY_VERSION=dev20250119
5555

5656
# Nightly version for torchvision
57-
VISION_NIGHTLY_VERSION=dev20241218
57+
VISION_NIGHTLY_VERSION=dev20250119
5858

5959
# Nightly version for torchtune
60-
TUNE_NIGHTLY_VERSION=dev20241218
60+
TUNE_NIGHTLY_VERSION=dev20250119
6161

6262
# The pip repository that hosts nightly torch packages. cpu by default.
6363
# If cuda is available, based on presence of nvidia-smi, install the pytorch nightly
@@ -79,15 +79,15 @@ fi
7979
if [[ -x "$(command -v xpu-smi)" ]];
8080
then
8181
REQUIREMENTS_TO_INSTALL=(
82-
torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
82+
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
8383
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
84-
torchtune=="0.5.0"
84+
torchtune=="0.6.0"
8585
)
8686
else
8787
REQUIREMENTS_TO_INSTALL=(
88-
torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
88+
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
8989
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
90-
torchtune=="0.5.0.${TUNE_NIGHTLY_VERSION}"
90+
torchtune=="0.6.0.${TUNE_NIGHTLY_VERSION}"
9191
)
9292
fi
9393

0 commit comments

Comments
 (0)