diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu
index 6226c127..8f91b9ba 100644
--- a/api/Containerfile.cuda.ubuntu
+++ b/api/Containerfile.cuda.ubuntu
@@ -12,14 +12,14 @@ ENV PATH="/onnx-web/onnx_web/bin:$PATH"
 
 RUN pip3 install pip --upgrade
 
+RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
+
 COPY requirements.txt /onnx-web/requirements.txt
 
 RUN pip3 install -r requirements.txt
 
 RUN pip3 install onnxruntime-gpu
 
-RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
-
 COPY onnx_web/ /onnx-web/onnx_web/
 
 ENTRYPOINT [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ]