fix flag

agirodi · Nov 23, 2023 · 17c90c9 · 17c90c9
1 parent 0b206f7
commit 17c90c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml
@@ -23,7 +23,7 @@ jobs:
         with:
           fallback: "dev"
       - name: Build Docker image GPU
-        run: python Docker/build.py --device gpu --tag girodia/fastsurfer:gpu-${{ steps.get_tag.outputs.tag }}
+        run: python Docker/build.py --device cuda --tag girodia/fastsurfer:gpu-${{ steps.get_tag.outputs.tag }}
       - name: Build Docker image CPU
         run: python Docker/build.py --device cpu --tag girodia/fastsurfer:cpu-${{ steps.get_tag.outputs.tag }}
       - name: Push Docker image GPU