diff --git a/docker/Dockerfile.tensorrt b/docker/Dockerfile.tensorrt
index 29ee1a4..de6027e 100644
--- a/docker/Dockerfile.tensorrt
+++ b/docker/Dockerfile.tensorrt
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
 
 ARG DEBIAN_FRONTEND=noninteractive