chore: bump cuda base version to a more recent one

Signed-off-by: SdgJlbl <[email protected]>
Substra · Sep 17, 2024 · 8551a2e · 8551a2e
1 parent 70c89f1
commit 8551a2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/changes/248.changed b/changes/248.changed
@@ -0,0 +1 @@
+cuda base docker image is now `nvidia/cuda:12.6.1-runtime-ubuntu24.04`
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
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		cuda base docker image is now `nvidia/cuda:12.6.1-runtime-ubuntu24.04`