df3934d
1
2
3
4
5
6
7
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime WORKDIR /app COPY requirements.txt . RUN pip install -r requirements.txt COPY . . EXPOSE 5000 CMD ["flask","run","--host=0.0.0.0"]