lym00 commited on
Commit
36244f1
·
verified ·
1 Parent(s): d4c0394

Upload AutoInstall Pytorch Triton Sage.bat

Browse files

allow manual pytorch nightly installation (useful for rollback)
```[1] Install PyTorch Nightly Latest (auto)
[2] Install PyTorch Nightly manually (specify version)
[3] Install PyTorch Stable
[0] Skip PyTorch installation
Choose PyTorch install option: 2

[INFO] Manual Nightly Installation
Visit the following URLs to find the latest compatible versions:
Torch: https://download.pytorch.org/whl/nightly/cu129/torch/
Torchaudio: https://download.pytorch.org/whl/nightly/cu129/torchaudio/
Torchvision:https://download.pytorch.org/whl/nightly/cu129/torchvision/

Enter base torch version (e.g. torch-2.9.0.dev20250715): torch-2.9.0.dev20250715
Enter base torchaudio version (e.g. torchaudio-2.8.0.dev20250716): torchaudio-2.8.0.dev20250716
Enter base torchvision version (e.g. torchvision-0.24.0.dev20250716): torchvision-0.24.0.dev20250716

[INFO] Installing manually specified wheels...
Torch: https://download.pytorch.org/whl/nightly/cu129/torch-2.9.0.dev20250715%2Bcu129-cp312-cp312-win_amd64.whl
Torchaudio: https://download.pytorch.org/whl/nightly/cu129/torchaudio-2.8.0.dev20250716%2Bcu129-cp312-cp312-win_amd64.whl
Torchvision:https://download.pytorch.org/whl/nightly/cu129/torchvision-0.24.0.dev20250716%2Bcu129-cp312-cp312-win_amd64.whl
Collecting torch==2.9.0.dev20250715+cu129

Files changed (1) hide show
  1. AutoInstall Pytorch Triton Sage.bat +53 -10
AutoInstall Pytorch Triton Sage.bat CHANGED
@@ -55,17 +55,27 @@ if /i "!inp!"=="y" set UPDATE_PIP=true
55
 
56
  REM Install PyTorch?
57
  set INSTALL_TORCH=false
 
 
58
  echo.
59
- echo [1] Install PyTorch Nightly
60
- echo [2] Install PyTorch Stable
 
61
  echo [0] Skip PyTorch installation
62
- set /p "torch_choice=Choose PyTorch version: "
 
63
  if "%torch_choice%"=="1" (
64
- set TORCH_NIGHTLY=true
65
  set INSTALL_TORCH=true
 
 
66
  ) else if "%torch_choice%"=="2" (
67
- set TORCH_NIGHTLY=false
68
  set INSTALL_TORCH=true
 
 
 
 
 
 
69
  )
70
 
71
  REM Install Triton?
@@ -109,12 +119,45 @@ if "%UPDATE_PIP%"=="true" (
109
  )
110
 
111
  if "%INSTALL_TORCH%"=="true" (
112
- echo [INFO] Uninstalling old PyTorch versions
113
- %PYTHON% -m pip uninstall -y torch torchvision torchaudio
114
-
115
  if "%TORCH_NIGHTLY%"=="true" (
116
- echo [INFO] Installing PyTorch Nightly for CUDA %CLEAN_CUDA%
117
- %PYTHON% -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu%CLEAN_CUDA%
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
118
  ) else (
119
  echo [INFO] Installing PyTorch Stable for CUDA %CLEAN_CUDA%
120
  %PYTHON% -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu%CLEAN_CUDA%
 
55
 
56
  REM Install PyTorch?
57
  set INSTALL_TORCH=false
58
+ set TORCH_NIGHTLY=false
59
+ set TORCH_MANUAL=false
60
  echo.
61
+ echo [1] Install PyTorch Nightly Latest (auto)
62
+ echo [2] Install PyTorch Nightly manually (specify version)
63
+ echo [3] Install PyTorch Stable
64
  echo [0] Skip PyTorch installation
65
+ set /p "torch_choice=Choose PyTorch install option: "
66
+
67
  if "%torch_choice%"=="1" (
 
68
  set INSTALL_TORCH=true
69
+ set TORCH_NIGHTLY=true
70
+ set TORCH_MANUAL=false
71
  ) else if "%torch_choice%"=="2" (
 
72
  set INSTALL_TORCH=true
73
+ set TORCH_NIGHTLY=true
74
+ set TORCH_MANUAL=true
75
+ ) else if "%torch_choice%"=="3" (
76
+ set INSTALL_TORCH=true
77
+ set TORCH_NIGHTLY=false
78
+ set TORCH_MANUAL=false
79
  )
80
 
81
  REM Install Triton?
 
119
  )
120
 
121
  if "%INSTALL_TORCH%"=="true" (
 
 
 
122
  if "%TORCH_NIGHTLY%"=="true" (
123
+ if "%TORCH_MANUAL%"=="true" (
124
+ echo.
125
+ echo [INFO] Manual Nightly Installation
126
+ echo Visit the following URLs to find the latest compatible versions:
127
+ echo Torch: https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/torch/
128
+ echo Torchaudio: https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/torchaudio/
129
+ echo Torchvision:https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/torchvision/
130
+ echo.
131
+
132
+ set /p "TORCH_VER=Enter base torch version (e.g. torch-2.9.0.dev20250715): "
133
+ set /p "AUDIO_VER=Enter base torchaudio version (e.g. torchaudio-2.8.0.dev20250716): "
134
+ set /p "VISION_VER=Enter base torchvision version (e.g. torchvision-0.24.0.dev20250716): "
135
+
136
+ for /f "tokens=1,2 delims=." %%a in ("%PYVER%") do (
137
+ set "TMP_PY_MAJ=%%a"
138
+ set "TMP_PY_MIN=%%b"
139
+ )
140
+
141
+ set "PY_MAJ=!TMP_PY_MAJ!"
142
+ set "PY_MIN=!TMP_PY_MIN!"
143
+ set "PY_TAG=cp!PY_MAJ!!PY_MIN!-cp!PY_MAJ!!PY_MIN!"
144
+ set "WHEEL_SUFFIX=%%2Bcu!CLEAN_CUDA!-!PY_TAG!-win_amd64.whl"
145
+
146
+ set "TORCH_URL=https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/!TORCH_VER!!WHEEL_SUFFIX!"
147
+ set "AUDIO_URL=https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/!AUDIO_VER!!WHEEL_SUFFIX!"
148
+ set "VISION_URL=https://download.pytorch.org/whl/nightly/cu!CLEAN_CUDA!/!VISION_VER!!WHEEL_SUFFIX!"
149
+
150
+ echo.
151
+ echo [INFO] Installing manually specified wheels...
152
+ echo Torch: !TORCH_URL!
153
+ echo Torchaudio: !AUDIO_URL!
154
+ echo Torchvision:!VISION_URL!
155
+
156
+ %PYTHON% -m pip install "!TORCH_URL!" "!AUDIO_URL!" "!VISION_URL!"
157
+ ) else (
158
+ echo [INFO] Installing PyTorch Nightly for CUDA %CLEAN_CUDA%
159
+ %PYTHON% -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu%CLEAN_CUDA%
160
+ )
161
  ) else (
162
  echo [INFO] Installing PyTorch Stable for CUDA %CLEAN_CUDA%
163
  %PYTHON% -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu%CLEAN_CUDA%