chore: bump cuda base version to a more recent one

Signed-off-by: SdgJlbl <[email protected]>
Substra · Sep 17, 2024 · 271b4a9 · 271b4a9
1 parent 70c89f1
commit 271b4a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/substrafl/remote/register/register.py b/substrafl/remote/register/register.py
@@ -37,7 +37,7 @@
 """
 
 _GPU_BASE_IMAGE = """
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM nvidia/cuda:12.6.1-runtime-ubuntu24.04
 
 # update image & install Python
 ENV DEBIAN_FRONTEND=noninteractive