From a9ce6fdac1e7eb758adf70a8b1a99e2833acdb5e Mon Sep 17 00:00:00 2001 From: SdgJlbl Date: Tue, 17 Sep 2024 17:04:52 +0200 Subject: [PATCH] chore: bump cuda base version to a more recent one Signed-off-by: SdgJlbl --- changes/248.changed | 1 + substrafl/remote/register/register.py | 2 +- tests/remote/register/test_register.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 changes/248.changed diff --git a/changes/248.changed b/changes/248.changed new file mode 100644 index 00000000..f41f5212 --- /dev/null +++ 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 index f7170421..03ec4c81 100644 --- 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 index 959eb014..5aa66658 100644 --- 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