Skip to content

Commit 56749b6

Browse files
authored
Bump PT/Vision/Tune Pin to 25250327 (#1516)
* [Test] Bump pin * Update install_requirements.sh * Update install_requirements.sh
1 parent dcc82b9 commit 56749b6

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

install/install_requirements.sh

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -51,20 +51,20 @@ 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=dev20250131
54+
PYTORCH_NIGHTLY_VERSION=dev20250327
5555

5656
# Nightly version for torchvision
57-
VISION_NIGHTLY_VERSION=dev20250131
57+
VISION_NIGHTLY_VERSION=dev20250327
5858

5959
# Nightly version for torchtune
60-
TUNE_NIGHTLY_VERSION=dev20250131
60+
TUNE_NIGHTLY_VERSION=dev20250327
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
6464
# with cuda for faster execution on cuda GPUs.
6565
if [[ -x "$(command -v nvidia-smi)" ]];
6666
then
67-
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu124"
67+
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu126"
6868
elif [[ -x "$(command -v rocminfo)" ]];
6969
then
7070
TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/rocm6.2"
@@ -79,15 +79,15 @@ fi
7979
if [[ -x "$(command -v xpu-smi)" ]];
8080
then
8181
REQUIREMENTS_TO_INSTALL=(
82-
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
82+
torch=="2.8.0.${PYTORCH_NIGHTLY_VERSION}"
8383
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
84-
#torchtune=="0.6.0" # no 0.6.0 on xpu nightly
84+
#torchtune=="0.7.0" # no 0.6.0 on xpu nightly
8585
)
8686
else
8787
REQUIREMENTS_TO_INSTALL=(
88-
torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}"
88+
torch=="2.8.0.${PYTORCH_NIGHTLY_VERSION}"
8989
torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}"
90-
torchtune=="0.6.0.${TUNE_NIGHTLY_VERSION}"
90+
torchtune=="0.7.0.${TUNE_NIGHTLY_VERSION}"
9191
)
9292
fi
9393

0 commit comments

Comments
 (0)