@@ -51,13 +51,13 @@ echo "Using pip executable: $PIP_EXECUTABLE"
51
51
# NOTE: If a newly-fetched version of the executorch repo changes the value of
52
52
# PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary
53
53
# package versions.
54
- PYTORCH_NIGHTLY_VERSION=dev20241218
54
+ PYTORCH_NIGHTLY_VERSION=dev20250119
55
55
56
56
# Nightly version for torchvision
57
- VISION_NIGHTLY_VERSION=dev20241218
57
+ VISION_NIGHTLY_VERSION=dev20250119
58
58
59
59
# Nightly version for torchtune
60
- TUNE_NIGHTLY_VERSION=dev20241218
60
+ TUNE_NIGHTLY_VERSION=dev20250119
61
61
62
62
# The pip repository that hosts nightly torch packages. cpu by default.
63
63
# If cuda is available, based on presence of nvidia-smi, install the pytorch nightly
79
79
if [[ -x " $( command -v xpu-smi) " ]];
80
80
then
81
81
REQUIREMENTS_TO_INSTALL=(
82
- torch==" 2.6 .0.${PYTORCH_NIGHTLY_VERSION} "
82
+ torch==" 2.7 .0.${PYTORCH_NIGHTLY_VERSION} "
83
83
torchvision==" 0.22.0.${VISION_NIGHTLY_VERSION} "
84
- torchtune==" 0.5 .0"
84
+ torchtune==" 0.6 .0"
85
85
)
86
86
else
87
87
REQUIREMENTS_TO_INSTALL=(
88
- torch==" 2.6 .0.${PYTORCH_NIGHTLY_VERSION} "
88
+ torch==" 2.7 .0.${PYTORCH_NIGHTLY_VERSION} "
89
89
torchvision==" 0.22.0.${VISION_NIGHTLY_VERSION} "
90
- torchtune==" 0.5 .0.${TUNE_NIGHTLY_VERSION} "
90
+ torchtune==" 0.6 .0.${TUNE_NIGHTLY_VERSION} "
91
91
)
92
92
fi
93
93
0 commit comments