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.

15 lines
302 B

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