File size: 237 Bytes
85cce87 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
WORKDIR /usr/app
ADD worker/requirements.txt .
RUN pip install -r requirements.txt
ADD wise .
WORKDIR /usr/app/worker
ADD worker/serve.py .
EXPOSE 8600
CMD ["python", "serve.py"] |