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"