Specify torchvision version

This commit is contained in:
oobabooga 2023-03-15 20:08:35 -03:00 committed by GitHub
parent 92d8b3e8dd
commit 4639bc84c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -17,10 +17,10 @@ set gpuchoice=!gpuchoice:a=A!
set gpuchoice=!gpuchoice:b=B! set gpuchoice=!gpuchoice:b=B!
if "%gpuchoice%" == "A" ( 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" set "CHANNEL=-c nvidia"
) else if "%gpuchoice%" == "B" ( ) 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=" set "CHANNEL="
) else ( ) else (
echo Invalid choice. Exiting... echo Invalid choice. Exiting...