diff --git a/ci/Dockerfile b/ci/Dockerfile
index 0673c89..447dd8b 100644
--- a/ci/Dockerfile
+++ b/ci/Dockerfile
@@ -3,5 +3,6 @@ FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install -y \
     python3-full \
+    python3-dev \
     git \
     libfftw3-dev