Merge pull request 'Docker + GPUs.' (#1) from mihasinkec/classla-api:master into master
Reviewed-on: #1master
commit
f406ccc9fc
@ -0,0 +1,11 @@
|
||||
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"]
|
Loading…
Reference in new issue