diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index fa169e7..4f24ff3 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -29,7 +29,7 @@ jobs:
       run:
         shell: bash -l {0}
     container:
-      image: openxrlab/xrsfm_runtime:ubuntu20.04_cu12.1.0
+      image: openxrlab/xrsfm_runtime:ubuntu20.04_cu11.6.1
       credentials:
         username: ${{secrets.DOCKERHUB_USERNAME}}
         password: ${{secrets.DOCKERHUB_PWD}}
diff --git a/Dockerfile b/Dockerfile
index bc1dff2..18091a1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu20.04
+FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive