@@ -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=dev20250324
5555
5656# Nightly version for torchvision
57- VISION_NIGHTLY_VERSION=dev20250131
57+ VISION_NIGHTLY_VERSION=dev20250324
5858
5959# Nightly version for torchtune
60- TUNE_NIGHTLY_VERSION=dev20250131
60+ TUNE_NIGHTLY_VERSION=dev20250324
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.
6565if [[ -x " $( command -v nvidia-smi) " ]];
6666then
67- TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/cu124 "
67+ TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/cu126 "
6868elif [[ -x " $( command -v rocminfo) " ]];
6969then
7070 TORCH_NIGHTLY_URL=" https://download.pytorch.org/whl/nightly/rocm6.2"
7979if [[ -x " $( command -v xpu-smi) " ]];
8080then
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} "
8484 # torchtune=="0.6.0" # no 0.6.0 on xpu nightly
8585 )
8686else
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} "
9090 torchtune==" 0.6.0.${TUNE_NIGHTLY_VERSION} "
9191 )
0 commit comments