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

schibsted · Sep 5, 2024 · eda7f1d · eda7f1d
1 parent 26752eb
commit eda7f1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.0.1-base-ubuntu22.04
+FROM nvidia/cuda:12.6.1-base-ubuntu22.04
 
 ENV PYTHON_VERSION=3.10