diff --git a/docker/Dockerfile b/docker/Dockerfile
index 95127b33a..2f7541c9a 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION=12.1.1
-FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu20.04
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
 ARG BUILD_TYPE=all
 ENV DEBIAN_FRONTEND=noninteractive