mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-31 22:50:15 +01:00
Specify torchvision version
This commit is contained in:
parent
92d8b3e8dd
commit
4639bc84c0
@ -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...
|
||||||
|
Loading…
Reference in New Issue
Block a user