From 86648d4085bec7b0aa3ebc0e1e5ced9b4f0c1782 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Fri, 22 Sep 2023 08:13:11 -0700 Subject: [PATCH] Remove CUDA, keep only pytorch --- CMD_FLAGS.txt | 1 + webui.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/CMD_FLAGS.txt b/CMD_FLAGS.txt index e69de29b..47fe8405 100644 --- a/CMD_FLAGS.txt +++ b/CMD_FLAGS.txt @@ -0,0 +1 @@ +--listen diff --git a/webui.py b/webui.py index aaa77885..4225cd7e 100644 --- a/webui.py +++ b/webui.py @@ -122,7 +122,7 @@ def install_dependencies(): # Install Pytorch if gpuchoice == "a": - run_cmd('conda install -y -k cuda ninja git -c nvidia/label/cuda-11.7.0 -c nvidia && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True) + run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True) elif gpuchoice == "b" and not is_macos(): if is_linux(): run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+rocm5.4.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2', assert_success=True, environment=True)