From 8f6d63331d7756ab4ecbafdd6701124e347983c6 Mon Sep 17 00:00:00 2001 From: Rob Ballantyne Date: Tue, 6 Aug 2024 17:05:53 +0100 Subject: [PATCH] Fix CUDA version --- .github/workflows/docker-build.yml | 2 +- build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml index 1aa6aef8..be5745ab 100644 --- a/.github/workflows/docker-build.yml +++ b/.github/workflows/docker-build.yml @@ -102,7 +102,7 @@ jobs: build: # Undeclared SHA tags with latest commit from master branch # Only building periodic sha tagged images - - {latest: "false", sha: "de17a97", python: "3.10", pytorch: "2.3.1", cuda: "12.1.0-base"} + - {latest: "false", sha: "de17a97", python: "3.10", pytorch: "2.3.1", cuda: "12.1.1-base"} steps: - name: Free Space diff --git a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh index a2a4f854..fe93b094 100755 --- a/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh +++ b/build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/cpu.sh @@ -6,7 +6,7 @@ build_cpu_main() { } build_cpu_install_deps() { - "$FOOOCUS_VENV_PIP" install --no-cache-dir \ + "$COMFYUI_VENV_PIP" install --no-cache-dir \ torch==${PYTORCH_VERSION} \ torchvision \ torchaudio \