diff --git a/install_pytorch.bat b/install_pytorch.bat index 0b38361..5fcf73f 100644 --- a/install_pytorch.bat +++ b/install_pytorch.bat @@ -1,6 +1,6 @@ @echo off -REM Install PyTorch, TorchVision, and Torchaudio with CUDA 11.8 support +REM Install PyTorch 2.5.0+cu118+cu118, TorchVision 0.20.0+cu118, and Torchaudio 2.5.0+cu118+cu118 with CUDA 11.8 support python --version >nul 2>&1 if %errorlevel% neq 0 ( @@ -10,7 +10,7 @@ if %errorlevel% neq 0 ( echo Installing PyTorch, TorchVision, and Torchaudio... -pip3 install torch==2.4.1+cu118 torchvision==0.19.1+cu118 torchaudio==2.4.1+cu118 --index-url https://download.pytorch.org/whl/cu118 +pip3 install torch==2.5.0+cu118 torchvision==0.20.0+cu118 torchaudio==2.5.0+cu118 --index-url https://download.pytorch.org/whl/cu118 if %errorlevel% neq 0 ( echo Installation failed.