1
0
Fork 0

fix(build): install torch before other packages in CPU container

This commit is contained in:
Sean Sube 2023-01-13 00:14:14 -06:00
parent fa639efff3
commit e025dbb87d
1 changed files with 3 additions and 3 deletions

View File

@ -10,11 +10,11 @@ ENV PATH="/onnx-web/onnx_web/bin:$PATH"
RUN pip3 install pip --upgrade
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu --no-cache-dir --ignore-installed
COPY requirements.txt /onnx-web/requirements.txt
RUN pip3 install -r requirements.txt
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip3 install -r requirements.txt --no-cache-dir
COPY onnx_web/ /onnx-web/onnx_web/