exadel-inc / exadel-inc/CompreFace

Update docker-compose.yml of GPU enabled builds so they also work on Docker Desktop Windows

Open
#1,103 1 comment 1 reaction 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.