fix(build): add cv2 deps to CUDA and ROCm images

ssube · Jan 21, 2023 · 52484e6 · 52484e6
1 parent a868c8c
commit 52484e6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu
@@ -1,10 +1,14 @@
 FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04
 
+WORKDIR /onnx-web/api
+
 RUN apt-get -y update \
- && apt-get -y install python3 python3-pip python3-venv \
+ && apt-get install -y ffmpeg libsm6 libxext6 \
  && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /onnx-web/api
+RUN apt-get -y update \
+ && apt-get -y install python3 python3-pip python3-venv \
+ && rm -rf /var/lib/apt/lists/*
 
 RUN python3 -m venv onnx_env
 

diff --git 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