apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(build): add venv to CPU image

This commit is contained in:
Sean Sube 2023-01-11 19:47:15 -06:00
parent 07c18c2245
commit 362b732707
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/Containerfile.cpu.buster
View File

@ -2,13 +2,19 @@ FROM docker.io/python:3.10-buster
WORKDIR /onnx-web
# TODO: set up venv within the container
RUN pip3 install virtualenv --upgrade
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
RUN python3 -m venv onnx_env
ENV PATH="/onnx-web/onnx_web/bin:$PATH"
RUN pip3 install pip --upgrade
COPY requirements.txt /onnx-web/requirements.txt
RUN pip install -r requirements.txt
RUN pip3 install -r requirements.txt
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
COPY onnx_web/ /onnx-web/onnx_web/

2
api/Containerfile.cuda.ubuntu
View File

@ -9,6 +9,8 @@ RUN python3 -m venv onnx_env
ENV PATH="/onnx-web/onnx_web/bin:$PATH"
RUN pip3 install pip --upgrade
COPY requirements.txt /onnx-web/requirements.txt
RUN pip3 install -r requirements.txt