fix: correct the cuda image version

gazai-io · Feb 12, 2025 · 5dd80a0 · 5dd80a0
1 parent c0fa6db
commit 5dd80a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
 # Stage 1: Base image with common dependencies
-FROM nvcr.io/nvidia/cuda:12.4.0-cudnn8-runtime-ubuntu22.04 as base
+FROM nvcr.io/nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 as base
 
 # Prevents prompts from packages asking for user input during installation
 ENV DEBIAN_FRONTEND=noninteractive