diff --git a/Dockerfile b/Dockerfile
index 6af04b3..4f6ae5f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -117,7 +117,7 @@ FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8
 
 RUN apt-get update && apt-get install -y --no-install-recommends gcc g++ \
-        python3 python3-pip python3-distutils ca-certificates \
+        python3 python3-dev python3-pip python3-distutils ca-certificates \
         libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 libjpeg8 && \
     rm -rf /var/lib/apt/lists/* && \
     python3 -m pip install --no-cache-dir --upgrade pip