Do not install CUDA in Windows if GPU is AMD

This commit is contained in:
Deniz Şafak
2025-05-20 02:48:02 +03:00
parent 93be50eda1
commit b3b420beec
+13 -3
View File
@@ -263,11 +263,18 @@ if "%MISAKI_LANG%" NEQ "en" (
) )
) )
:: Check GPU vendor
echo Checking GPU vendor...
set GPU_VENDOR=unknown
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from abogen.utils import get_gpu_vendor; print(get_gpu_vendor())"') do set GPU_VENDOR=%%i
echo GPU Vendor: %GPU_VENDOR%
:: Check if torch is installed with CUDA support :: Check if torch is installed with CUDA support
echo Checking CUDA availability... echo Checking CUDA availability...
for /f %%i in ('%PYTHON_CONSOLE_PATH% -c "from torch.cuda import is_available; print(is_available())"') do set cuda_available=%%i if /I "%GPU_VENDOR%"=="nvidia" (
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 %CUDA_VERSION% 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 %PYTHON_CONSOLE_PATH% -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu%CUDA_VERSION% --no-warn-script-location
echo. echo.
@@ -276,8 +283,11 @@ if "%cuda_available%"=="False" (
pause pause
exit /b exit /b
) )
) else (
echo CUDA is available on NVIDIA GPU.
)
) else ( ) else (
echo CUDA is available. echo Non-NVIDIA GPU detected (%GPU_VENDOR%). Skipping PyTorch CUDA installation.
) )
:: Ask user if they want to create a desktop shortcut :: Ask user if they want to create a desktop shortcut