-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdocker-compose.yml
More file actions
29 lines (25 loc) · 912 Bytes
/
docker-compose.yml
File metadata and controls
29 lines (25 loc) · 912 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
version: "3.9"
x-base: &egra_base
image: nemo_inference_nemo-asr:latest
build:
context: .
dockerfile: docker/Dockerfile
working_dir: /work
environment:
PYTHONUNBUFFERED: "1"
volumes:
- ./:/work:rw # your repo root (contains main.py, infer.py, etc.)
- ./input_output_data:/io:rw # input/output data root
- ./nemo_inference/models:/models:ro # .nemo models
- ./nemo_inference/tmp:/tmp_segments:rw # temp segments (and kept if debug flag is used)
services:
nemo-asr:
<<: *egra_base
container_name: nemo-asr
# GPU access is disabled by default to avoid nvidia-container errors on hosts
# without a properly configured NVIDIA driver / toolkit. To enable GPU, uncomment:
# gpus: "all"
egra-eval:
<<: *egra_base
container_name: egra-eval
# gpus: "all" # not needed for CPU-only eval