mihasinkec/classla-api
1
0
Fork 0
forked from lkrsnik/classla-api
Code Issues Pull Requests Projects Releases Wiki Activity

Docker + GPUs.

Browse Source
This commit is contained in:
Mihael Šinkec 2022-01-18 10:08:59 +01:00
parent f75bd28512
commit f75a6a6a9f
2 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Dockerfile.gpu Normal file
View File

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

14
docker-compose.yml
View File

@ -1,15 +1,13 @@
version: "3.8" version: "3.7"
services: services:
app: app:
build: . build: .
command: python -m flask run --host=0.0.0.0 command: python -m flask run --host=0.0.0.0
deploy: runtime: nvidia
resources: environment:
reservations: - NVIDIA_VISIBLE_DEVICES=all
devices: - NVIDIA_DRIVER_CAPABILITIES=all
- driver: nvidia - CUDA_VISIBLE_DEVICES=1
count: all
capabilities: [ gpu ]
ports: ports:
- "5000:5000" - "5000:5000"
volumes: volumes: