coco-annotator icon indicating copy to clipboard operation
coco-annotator copied to clipboard

no gpu docker image

Open fenghuoxiguozu opened this issue 3 years ago • 1 comments

image

fenghuoxiguozu avatar Jul 20 '22 07:07 fenghuoxiguozu

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.

SixK avatar Jun 09 '23 09:06 SixK