chore(deps): update nvidia/cuda docker tag to v12.3.2

billsioros · Feb 28, 2024 · 8a74cd4 · 8a74cd4
1 parent e9c5116
commit 8a74cd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile.train b/docker/Dockerfile.train
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.3.1-runtime-ubuntu22.04 as cuda-base
+FROM nvidia/cuda:12.3.2-runtime-ubuntu22.04 as cuda-base
 
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]