diff --git a/.github/scripts/DockerfileGenerator.sh b/.github/scripts/DockerfileGenerator.sh
index 1849333..bf65d8b 100755
--- a/.github/scripts/DockerfileGenerator.sh
+++ b/.github/scripts/DockerfileGenerator.sh
@@ -206,7 +206,7 @@ elif [ "${COMPILER_NAME}" = 'cuda' ]
 then
   C_COMPILER='gcc'
   CXX_COMPILER='g++'
-  CMAKE_CONFIG+='-D ARCANE_ACCELERATOR_MODE=CUDA -D CMAKE_CUDA_COMPILER=nvcc '
+  CMAKE_CONFIG+='-D ARCANE_ACCELERATOR_MODE=CUDA -D CMAKE_CUDA_COMPILER=nvcc -D CMAKE_CUDA_ARCHITECTURES="80;90" '
 
 else
   echo "Unknown compiler: ${COMPILER_NAME}"