diff --git a/dockerfile b/dockerfile
index c645e79..9cfe5f1 100644
--- a/dockerfile
+++ b/dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.4.1-base-ubuntu22.04
+FROM nvidia/cuda:12.6.3-base-ubuntu22.04
 
 ENV PYTHON_VERSION=3.11