Upload AutoInstall Pytorch Triton Sage.bat
Browse filesallow 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
@@ -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
|
|
|
61 |
echo [0] Skip PyTorch installation
|
62 |
-
set /p "torch_choice=Choose PyTorch
|
|
|
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 |
-
|
117 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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%
|