Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 17:50:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use the official instructions

Browse Source 
https://pytorch.org/get-started/locally/
This commit is contained in:
oobabooga 2023-03-17 20:37:25 -03:00 committed by GitHub
parent 4639bc84c0
commit 91371640f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
install.bat
View File

@ -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...