mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-31 22:50:15 +01:00
Remove explicit pytorch installation
Fixes an issue some people were having: https://github.com/oobabooga/text-generation-webui/issues/15 I did not experience this issue on my system. Not everyone does for some reason.
This commit is contained in:
parent
6f89242094
commit
bdf85ffcf9
@ -14,7 +14,7 @@ set /p "gpuchoice=Input> "
|
|||||||
set gpuchoice=%gpuchoice:~0,1%
|
set gpuchoice=%gpuchoice:~0,1%
|
||||||
|
|
||||||
if /I "%gpuchoice%" == "A" (
|
if /I "%gpuchoice%" == "A" (
|
||||||
set "PACKAGES_TO_INSTALL=python=3.10.9 pytorch torchvision torchaudio pytorch-cuda=11.7 cuda-toolkit conda-forge::ninja conda-forge::git"
|
set "PACKAGES_TO_INSTALL=python=3.10.9 torchvision torchaudio pytorch-cuda=11.7 cuda-toolkit conda-forge::ninja conda-forge::git"
|
||||||
set "CHANNEL=-c pytorch -c nvidia/label/cuda-11.7.0 -c nvidia"
|
set "CHANNEL=-c pytorch -c nvidia/label/cuda-11.7.0 -c nvidia"
|
||||||
) else if /I "%gpuchoice%" == "B" (
|
) else if /I "%gpuchoice%" == "B" (
|
||||||
set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio cpuonly git"
|
set "PACKAGES_TO_INSTALL=pytorch torchvision torchaudio cpuonly git"
|
||||||
|
Loading…
Reference in New Issue
Block a user