apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-01-13 00:38:36 -06:00
parent a2c269e360
commit 8fff742443
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/Containerfile.cuda.ubuntu
View File

@ -12,14 +12,14 @@ ENV PATH="/onnx-web/onnx_web/bin:$PATH"
RUN pip3 install pip --upgrade RUN pip3 install pip --upgrade
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
COPY requirements.txt /onnx-web/requirements.txt COPY requirements.txt /onnx-web/requirements.txt
RUN pip3 install -r requirements.txt RUN pip3 install -r requirements.txt
RUN pip3 install onnxruntime-gpu RUN pip3 install onnxruntime-gpu
RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117
COPY onnx_web/ /onnx-web/onnx_web/ COPY onnx_web/ /onnx-web/onnx_web/
ENTRYPOINT [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ] ENTRYPOINT [ "flask", "--app=onnx_web.serve", "run", "--host=0.0.0.0" ]