15 lines
302 B
YAML
15 lines
302 B
YAML
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
|