From 9ed3a03d4b77591e50626818ba7238058ca6444a Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 18 Mar 2023 11:25:08 -0300
Subject: [PATCH] Don't use the official instructions

---
 install.bat | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/install.bat b/install.bat
index 8d923589..4dc7018c 100644
--- a/install.bat
+++ b/install.bat
@@ -17,7 +17,7 @@ set gpuchoice=!gpuchoice:a=A!
 set gpuchoice=!gpuchoice:b=B!
 
 if "%gpuchoice%" == "A" (
-    set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio pytorch-cuda=11.7 conda git"
+    set "PACKAGES_TO_INSTALL=torchvision=0.14.1 torchaudio=0.13.1 pytorch-cuda=11.7 conda git"
     set "CHANNEL=-c nvidia"
 ) else if "%gpuchoice%" == "B" (
     set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio cpuonly conda git"