From 52484e6e1f85973ec2eeb69cd64288340a814ad6 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 21 Jan 2023 17:01:02 -0600 Subject: [PATCH] fix(build): add cv2 deps to CUDA and ROCm images --- api/Containerfile.cuda.ubuntu | 8 ++++++-- api/Containerfile.rocm.ubuntu | 4 ++++ 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu index 04713332..9e9bf7a0 100644 --- a/api/Containerfile.cuda.ubuntu +++ b/api/Containerfile.cuda.ubuntu @@ -1,11 +1,15 @@ FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04 +WORKDIR /onnx-web/api + +RUN apt-get -y update \ + && apt-get install -y ffmpeg libsm6 libxext6 \ + && rm -rf /var/lib/apt/lists/* + RUN apt-get -y update \ && apt-get -y install python3 python3-pip python3-venv \ && rm -rf /var/lib/apt/lists/* -WORKDIR /onnx-web/api - RUN python3 -m venv onnx_env ENV PATH="/onnx-web/api/onnx_web/bin:$PATH" diff --git a/api/Containerfile.rocm.ubuntu b/api/Containerfile.rocm.ubuntu index 9adf9e8b..a62995db 100644 --- a/api/Containerfile.rocm.ubuntu +++ b/api/Containerfile.rocm.ubuntu @@ -2,6 +2,10 @@ FROM docker.io/rocm/pytorch:rocm5.4_ubuntu20.04_py3.8_pytorch_1.12.1 WORKDIR /onnx-web/api +RUN apt-get -y update \ + && apt-get install -y ffmpeg libsm6 libxext6 \ + && rm -rf /var/lib/apt/lists/* + RUN pip3 install virtualenv --upgrade RUN python3 -m venv onnx_env