Merge pull request #14 from szmazurek/feature/torch_2_1_0

Hotfix MLcube config
szmazurek · Nov 25, 2023 · 3fcfe84 · 3fcfe84
2 parents 01638d6 + b8dba8a
commit 3fcfe84
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: