diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index 85d8052f..b7b6db83 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -16,7 +16,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", comfyui: "v0.2.0", python: "3.10", pytorch: "2.4.0"}
+          - {latest: "false", comfyui: "v0.2.2", python: "3.10", pytorch: "2.4.1"}
     steps:
       -
         name: Free Space
@@ -98,7 +98,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", comfyui: "v0.2.0", python: "3.10", pytorch: "2.3.1", cuda: "12.1.1-base"}
+          - {latest: "false", comfyui: "v0.2.2", python: "3.10", pytorch: "2.4.1", cuda: "12.1.1-base"}
     steps:
       -
         name: Free Space
@@ -183,7 +183,7 @@ jobs:
       fail-fast: false
       matrix:
         build:
-          - {latest: "true", comfyui: "v0.2.0", python: "3.10", pytorch: "2.3.1", rocm: "6.0-runtime"}
+          - {latest: "false", comfyui: "v0.2.2", python: "3.10", pytorch: "2.3.1", rocm: "6.0-runtime"}
     steps:
       -
         name: Free Space
diff --git a/docker-compose.yaml b/docker-compose.yaml
index de4bcdd9..981edaf4 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -7,7 +7,7 @@ services:
       context: ./build
       args:
         PYTHON_VERSION: ${PYTHON_VERSION:-3.10}
-        PYTORCH_VERSION: ${PYTORCH_VERSION:-2.3.0}
+        PYTORCH_VERSION: ${PYTORCH_VERSION:-2.4.1}
         COMFYUI_BUILD_REF: ${COMFYUI_BUILD_REF:-}
         # Base on Python image which is base + python + jupyter
         IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:${PYTHON_VERSION:-3.10}-v2-cuda-12.1.1-base-22.04}