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

dreamquark-ai · Jul 13, 2024 · 9a14ab5 · 9a14ab5
1 parent 2c0c4eb
commit 9a14ab5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile_gpu b/Dockerfile_gpu
@@ -1,5 +1,5 @@
 # GENERATED FROM SCRIPTS
-FROM nvidia/cuda:11.3.1-runtime-ubuntu20.04
+FROM nvidia/cuda:12.5.1-runtime-ubuntu20.04@sha256:166a06a0c99fbf96de08451b606c1e8a8366000105dc632f4c7cf1bcb6891827
 
 # Avoid tzdata interactive action
 ENV DEBIAN_FRONTEND noninteractive