@@ -51,20 +51,20 @@ 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=dev20250131
54
+ PYTORCH_NIGHTLY_VERSION=dev20250327
55
55
56
56
# Nightly version for torchvision
57
- VISION_NIGHTLY_VERSION=dev20250131
57
+ VISION_NIGHTLY_VERSION=dev20250327
58
58
59
59
# Nightly version for torchtune
60
- TUNE_NIGHTLY_VERSION=dev20250131
60
+ TUNE_NIGHTLY_VERSION=dev20250327
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
64
64
# with cuda for faster execution on cuda GPUs.
65
65
if [[ -x " $( command -v nvidia-smi) " ]];
66
66
then
67
- TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/cu124 "
67
+ TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/cu126 "
68
68
elif [[ -x " $( command -v rocminfo) " ]];
69
69
then
70
70
TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/rocm6.2"
79
79
if [[ -x " $( command -v xpu-smi) " ]];
80
80
then
81
81
REQUIREMENTS_TO_INSTALL=(
82
- torch==" 2.7 .0.${PYTORCH_NIGHTLY_VERSION} "
82
+ torch==" 2.8 .0.${PYTORCH_NIGHTLY_VERSION} "
83
83
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
85
85
)
86
86
else
87
87
REQUIREMENTS_TO_INSTALL=(
88
- torch==" 2.7 .0.${PYTORCH_NIGHTLY_VERSION} "
88
+ torch==" 2.8 .0.${PYTORCH_NIGHTLY_VERSION} "
89
89
torchvision==" 0.22.0.${VISION_NIGHTLY_VERSION} "
90
- torchtune==" 0.6 .0.${TUNE_NIGHTLY_VERSION} "
90
+ torchtune==" 0.7 .0.${TUNE_NIGHTLY_VERSION} "
91
91
)
92
92
fi
93
93
0 commit comments