You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

12 lines
260 B

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"]