Fix CUDA version

ai-dock · Aug 6, 2024 · 8f6d633 · 8f6d633
1 parent ac5bf6f
commit 8f6d633
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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 \