fix(build): add cv2 deps to CUDA and ROCm images
This commit is contained in:
parent
a868c8cf6b
commit
52484e6e1f
|
@ -1,11 +1,15 @@
|
||||||
FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04
|
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 \
|
RUN apt-get -y update \
|
||||||
&& apt-get -y install python3 python3-pip python3-venv \
|
&& apt-get -y install python3 python3-pip python3-venv \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
WORKDIR /onnx-web/api
|
|
||||||
|
|
||||||
RUN python3 -m venv onnx_env
|
RUN python3 -m venv onnx_env
|
||||||
|
|
||||||
ENV PATH="/onnx-web/api/onnx_web/bin:$PATH"
|
ENV PATH="/onnx-web/api/onnx_web/bin:$PATH"
|
||||||
|
|
|
@ -2,6 +2,10 @@ FROM docker.io/rocm/pytorch:rocm5.4_ubuntu20.04_py3.8_pytorch_1.12.1
|
||||||
|
|
||||||
WORKDIR /onnx-web/api
|
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 pip3 install virtualenv --upgrade
|
||||||
|
|
||||||
RUN python3 -m venv onnx_env
|
RUN python3 -m venv onnx_env
|
||||||
|
|
Loading…
Reference in New Issue