Update install_pytorch.bat

Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
FNBUBBLES420-ORG · Sep 6, 2024 · 8225752 · 8225752
1 parent 7f96bc0
commit 8225752
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/install_pytorch.bat b/install_pytorch.bat
@@ -1,14 +1,16 @@
 @echo off
 
+REM Install PyTorch, TorchVision, and Torchaudio with CUDA 11.8 support
+
 python --version >nul 2>&1
 if %errorlevel% neq 0 (
     echo Python is not installed. Please install Python and try again.
     exit /b 1
 )
 
-echo Installing PyTorch and related packages...
+echo Installing PyTorch, TorchVision, and Torchaudio...
 
-pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 if %errorlevel% neq 0 (
     echo Installation failed.