Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Modify PyTorch installation for CUDA 12.8

Browse Source 
Updated PyTorch installation commands to use version 2.8.0 with CUDA (12.8) support.
This commit is contained in:
Deniz Şafak
2025-10-30 12:06:24 +03:00
committed by GitHub
parent 42b7979585
commit 334c0654ca
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
WINDOWS_INSTALL.bat
+6 -7
View File
@@ -8,9 +8,6 @@ cd /d "%~dp0"
:: Japanese: "ja" :: Japanese: "ja"
set MISAKI_LANG=en set MISAKI_LANG=en
:: Set PyTorch CUDA version
set CUDA_VERSION=128
::: :::
::: _ ____ ___ ____ _____ _ _ ::: _ ____ ___ ____ _____ _ _
::: / \ | __ ) / _ \ / ___|| ____| \ | | ::: / \ | __ ) / _ \ / ___|| ____| \ | |
@@ -281,8 +278,9 @@ if /I "%IS_NVIDIA%"=="true" (
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from torch.cuda import is_available; print(is_available())"') do set cuda_available=%%i for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from torch.cuda import is_available; print(is_available())"') do set cuda_available=%%i
if "%cuda_available%"=="False" ( if "%cuda_available%"=="False" (
echo Installing PyTorch with CUDA %CUDA_VERSION% support... echo Installing PyTorch with CUDA (12.8) support...
%PYTHON_CONSOLE_PATH% -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu%CUDA_VERSION% --no-warn-script-location :: Temporarily set the PyTorch version to 2.0.8 to prevent [WinError 1114] error.
%PYTHON_CONSOLE_PATH% -m pip install torch==2.8.0 torchaudio==2.8.0 torchvision==0.23.0 --extra-index-url https://download.pytorch.org/whl/cu128 --no-warn-script-location
echo. echo.
if errorlevel 1 ( if errorlevel 1 (
echo Failed to install PyTorch. echo Failed to install PyTorch.
@@ -305,8 +303,9 @@ if /I "%IS_NVIDIA%"=="true" (
if errorlevel 2 ( if errorlevel 2 (
echo Skipping PyTorch installation. echo Skipping PyTorch installation.
) else ( ) else (
echo Installing PyTorch with CUDA %CUDA_VERSION% support... echo Installing PyTorch with CUDA (12.8) support...
%PYTHON_CONSOLE_PATH% -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu%CUDA_VERSION% --no-warn-script-location :: Temporarily set the PyTorch version to 2.0.8 to prevent [WinError 1114] error.
%PYTHON_CONSOLE_PATH% -m pip install torch==2.8.0 torchaudio==2.8.0 torchvision==0.23.0 --extra-index-url https://download.pytorch.org/whl/cu128 --no-warn-script-location
if errorlevel 1 ( if errorlevel 1 (
echo Failed to install PyTorch. echo Failed to install PyTorch.
pause pause