From 6df895151830955d86be129c1549a8456a674fa1 Mon Sep 17 00:00:00 2001
From: "renovate[bot]" <29139614+renovate[bot]@users.noreply.github.com>
Date: Thu, 11 Apr 2024 00:00:49 +0000
Subject: [PATCH] chore(deps): update nvidia/cuda docker tag to v12.4.1

---
 docker/Dockerfile.train | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile.train b/docker/Dockerfile.train
index 76fe695..5294894 100644
--- a/docker/Dockerfile.train
+++ b/docker/Dockerfile.train
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.0-runtime-ubuntu22.04 as cuda-base
+FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04 as cuda-base
 
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]