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

Merge pull request 'Docker + GPUs.' (#1) from mihasinkec/classla-api:master into master

Browse Source 
Reviewed-on: lkrsnik/classla-api#1
This commit is contained in:
lkrsnik 2022-01-18 15:00:53 +00:00
commit f406ccc9fc
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: