diff --git a/docker/Dockerfile b/docker/Dockerfile
index 9617371..96607ee 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 # Use an official CUDA runtime with Ubuntu as a parent image
-FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
+FROM nvidia/cuda:12.8.1-runtime-ubuntu24.04
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \