Dockerfile

FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime

WORKDIR /app

COPY requirements.txt .
RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt

COPY *.py .

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]