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