diff --git a/build/linux/Dockerfile b/build/linux/Dockerfile
index d432bed..00dc22d 100644
--- a/build/linux/Dockerfile
+++ b/build/linux/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
 WORKDIR /app
 ARG DEBIAN_FRONTEND=noninteractive
 RUN apt update