diff --git a/Dockerfile b/Dockerfile
index 9eb1037..8a8082b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -28,7 +28,7 @@ COPY python/ ./python/
 RUN npm run build
 
 # Stage 2: Production runtime
-FROM nvidia/cuda:11.8-runtime-ubuntu22.04
+FROM nvcr.io/nvidia/cuda:13.0.1-runtime-ubuntu22.04
 
 # Install runtime dependencies
 RUN apt-get update && apt-get install -y \