From 91371640f944cb505def354a51483d9f834bf649 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Fri, 17 Mar 2023 20:37:25 -0300
Subject: [PATCH] Use the official instructions

https://pytorch.org/get-started/locally/
---
 install.bat | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/install.bat b/install.bat
index 768ebf8b..8d923589 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=0.14.1 torchaudio=0.13.1 pytorch-cuda=11.7 conda git"
+    set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio pytorch-cuda=11.7 conda git"
     set "CHANNEL=-c nvidia"
 ) else if "%gpuchoice%" == "B" (
-    set "PACKAGES_TO_INSTALL=torchvision=0.14.1 torchaudio=0.13.1 pytorch conda git"
+    set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio cpuonly conda git"
     set "CHANNEL="
 ) else (
     echo Invalid choice. Exiting...