chore: bump cuda base version to a more recent one

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

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
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

diff --git a/tests/remote/register/test_register.py b/tests/remote/register/test_register.py
@@ -56,7 +56,7 @@ def test_get_base_docker_image_cpu():
 
 def test_get_base_docker_image_gpu():
     expected_dockerfile = """
-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`