diff --git a/.github/workflows/build-and-push-dockerimage.yml b/.github/workflows/build-and-push-dockerimage.yml
index 22ef30a..648e01b 100644
--- a/.github/workflows/build-and-push-dockerimage.yml
+++ b/.github/workflows/build-and-push-dockerimage.yml
@@ -48,7 +48,7 @@ jobs:
       uses: docker/build-push-action@v2
       with:
         context: .
-        file: ./docker/env.Dockerfile
+        file: ./docker/env.cuda.Dockerfile  # Updated for GPU build
         platforms: linux/amd64
         push: true
         tags: |
@@ -59,7 +59,6 @@ jobs:
         build-args: |
           BASE_IMAGE=${{ env.BASE_IMAGE }}
 
-
   build-nogpu:
     name: Build and push docker image cpu 
     runs-on: ubuntu-latest
@@ -82,7 +81,7 @@ jobs:
         echo "IMAGE_NAME=$image_name" >> $GITHUB_ENV
         echo "TAG=$tag" >> $GITHUB_ENV
         echo "BASE_IMAGE=$base_image" >> $GITHUB_ENV
-        echo "FRAMEWORK=$framework" >> $GITHUB_ENV0
+        echo "FRAMEWORK=$framework" >> $GITHUB_ENV
 
     - name: Set up Docker Buildx
       uses: docker/setup-buildx-action@v1
@@ -104,7 +103,7 @@ jobs:
       uses: docker/build-push-action@v2
       with:
         context: .
-        file: ./docker/env.Dockerfile
+        file: ./docker/env.cpu.Dockerfile  # Updated for CPU build
         platforms: linux/amd64
         push: true
         tags: |
@@ -114,3 +113,4 @@ jobs:
         cache-to: type=gha,mode=max
         build-args: |
           BASE_IMAGE=${{ env.BASE_IMAGE }}
+