12 lines
260 B
Docker
12 lines
260 B
Docker
|
FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime
|
||
|
|
|
||
|
|
ENV PYTHONDONTWRITEBYTECODE 1
|
||
|
|
ENV PYTHONUNBUFFERED 1
|
||
|
|
|
||
|
|
ADD . /classla-api
|
||
|
|
WORKDIR /classla-api
|
||
|
|
|
||
|
|
RUN pip install $(grep -ivE "torch" requirements.txt)
|
||
|
|
|
||
|
|
CMD [ "python3", "-m" , "flask", "run", "--host=0.0.0.0"]
|