exadel-inc / exadel-inc/CompreFace
Update docker-compose.yml of GPU enabled builds so they also work on Docker Desktop Windows
- Dominant language
- Java
- Stars
- 8.3k
- Forks
- 1.2k
- PR merge metrics
- No merged PRs in 30d
Description
**Is your feature request related to a problem? Please describe.**
Current use of `runtime: nvidia` in the `compreface-core:` service of ` docker-compose.yml` for GPU enabled builds means that they will not work on Docker Desktop Windows (At least I have not being able to find a solution on how to install the Nvidia runtime on Windows)
**Describe the solution you'd like**
Switch to enabling GPU access using the "device" structure as detailed in https://docs.docker.com/compose/gpu-support/ This works on Windows and I presume it also works on Docker Linux
**Additional context**
See: https://github.com/exadel-inc/CompreFace/discussions/1084#discussioncomment-6408348 for proof that this works
Example:
```
compreface-core:
image: ${registry}compreface-core:${CORE_VERSION}
restart: always
container_name: "compreface-core"
environment:
- ML_PORT=3000
- UWSGI_PROCESSES=${uwsgi_processes:-2}
- UWSGI_THREADS=${uwsgi_threads:-1}
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
```
Contributor guide
Assessment
This issue has not been assessed yet.