diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu
index 427d664b..6226c127 100644
--- a/api/Containerfile.cuda.ubuntu
+++ b/api/Containerfile.cuda.ubuntu
@@ -1,7 +1,8 @@
 FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04
 
-RUN apt -y update \
- && apt -y install python3 python3-pip python3-venv
+RUN apt-get -y update \
+ && apt-get -y install python3 python3-pip python3-venv \
+ && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /onnx-web