version: "3.7"
services:
app:
build: .
command: python -m flask run --host=0.0.0.0
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=all
- CUDA_VISIBLE_DEVICES=1
ports:
- "5000:5000"
volumes:
- .:/classla-api