coco-annotator
coco-annotator copied to clipboard
no gpu docker image

I think you have to run the following script before : https://github.com/jsbroks/coco-annotator/blob/master/build_gpu.sh
Note, that I'm not sure the docker-compose.gpu.yml really use GPU by default.
I had to activate it adding the following lines: `
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: [gpu]
`
See this commit: https://github.com/SixK/coco-annotator/commit/83226e8e9a618141797525cdf2d5afbf554c2b6a
Note you will also have to configure docker to use GPU in /etc/docker/daemon.json: { "default-runtime": "nvidia", "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } }, }
maybe I didn't configured this line on my system "default-runtime": "nvidia", but I don't think so.