From 271b4a949ded6a6c9e199ce494e16081c3df67cc Mon Sep 17 00:00:00 2001
From: SdgJlbl <sarah.diot-girard@owkin.com>
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 <sarah.diot-girard@owkin.com>
---
 substrafl/remote/register/register.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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