Hotfix MLcube config

szmazurek · Nov 25, 2023 · b8dba8a · b8dba8a
1 parent 99ee027
commit b8dba8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlcube/model_mlcube/mlcube.yaml b/mlcube/model_mlcube/mlcube.yaml
@@ -30,7 +30,7 @@ docker:
   # Docker build context relative to $MLCUBE_ROOT. (gandlf_deploy can handle this automatically.)
   build_context: "../"
   # Docker file name within docker build context. Any "Dockerfile-*" in the GaNDLF source repo is valid.
-  build_file: "Dockerfile-CUDA11.6"
+  build_file: "Dockerfile-CUDA11.8"
 
   # These settings should be set by global MLCube configuration, generally not per-deployment.
   # However, some sane defaults (for Docker >19.03) are here: