diff --git a/install/install_requirements.sh b/install/install_requirements.sh index 0821d5585..779969cac 100755 --- a/install/install_requirements.sh +++ b/install/install_requirements.sh @@ -51,20 +51,20 @@ echo "Using pip executable: $PIP_EXECUTABLE" # NOTE: If a newly-fetched version of the executorch repo changes the value of # PYTORCH_NIGHTLY_VERSION, you should re-run this script to install the necessary # package versions. -PYTORCH_NIGHTLY_VERSION=dev20250131 +PYTORCH_NIGHTLY_VERSION=dev20250327 # Nightly version for torchvision -VISION_NIGHTLY_VERSION=dev20250131 +VISION_NIGHTLY_VERSION=dev20250327 # Nightly version for torchtune -TUNE_NIGHTLY_VERSION=dev20250131 +TUNE_NIGHTLY_VERSION=dev20250327 # The pip repository that hosts nightly torch packages. cpu by default. # If cuda is available, based on presence of nvidia-smi, install the pytorch nightly # with cuda for faster execution on cuda GPUs. if [[ -x "$(command -v nvidia-smi)" ]]; then - TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu124" + TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/cu126" elif [[ -x "$(command -v rocminfo)" ]]; then TORCH_NIGHTLY_URL="https://download.pytorch.org/whl/nightly/rocm6.2" @@ -79,15 +79,15 @@ fi if [[ -x "$(command -v xpu-smi)" ]]; then REQUIREMENTS_TO_INSTALL=( - torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}" + torch=="2.8.0.${PYTORCH_NIGHTLY_VERSION}" torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}" - #torchtune=="0.6.0" # no 0.6.0 on xpu nightly + #torchtune=="0.7.0" # no 0.6.0 on xpu nightly ) else REQUIREMENTS_TO_INSTALL=( - torch=="2.7.0.${PYTORCH_NIGHTLY_VERSION}" + torch=="2.8.0.${PYTORCH_NIGHTLY_VERSION}" torchvision=="0.22.0.${VISION_NIGHTLY_VERSION}" - torchtune=="0.6.0.${TUNE_NIGHTLY_VERSION}" + torchtune=="0.7.0.${TUNE_NIGHTLY_VERSION}" ) fi