apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-01-21 17:01:02 -06:00
parent a868c8cf6b
commit 52484e6e1f
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/Containerfile.cuda.ubuntu
View File

@ -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"

4
api/Containerfile.rocm.ubuntu
View File

@ -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