From 4639bc84c05cf6f9f452b3a7351827215fc55477 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 15 Mar 2023 20:08:35 -0300
Subject: [PATCH] Specify torchvision version

---
 install.bat | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/install.bat b/install.bat
index 963c1ce2..768ebf8b 100644
--- a/install.bat
+++ b/install.bat
@@ -17,10 +17,10 @@ set gpuchoice=!gpuchoice:a=A!
 set gpuchoice=!gpuchoice:b=B!
 
 if "%gpuchoice%" == "A" (
-    set "PACKAGES_TO_INSTALL=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=torchvision torchaudio pytorch conda git"
+    set "PACKAGES_TO_INSTALL=torchvision=0.14.1 torchaudio=0.13.1 pytorch conda git"
     set "CHANNEL="
 ) else (
     echo Invalid choice. Exiting...