fix(build): use CPU version of pytorch in hypothetical alpine API image
This commit is contained in:
parent
ca0da31791
commit
af40bff10c
|
@ -10,6 +10,6 @@ COPY serve.py /onnx-web/serve.py
|
||||||
|
|
||||||
RUN pip install -r requirements.txt
|
RUN pip install -r requirements.txt
|
||||||
|
|
||||||
# RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
|
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
|
||||||
|
|
||||||
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]
|
ENTRYPOINT [ "flask", "--app=serve", "run", "--host=0.0.0.0" ]
|
||||||
|
|
Loading…
Reference in New Issue