8000 TF_annotation does not use a GPU with the CUDA component installed. · Issue #1124 · cvat-ai/cvat · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
TF_annotation does not use a GPU with the CUDA component installed. #1124
Closed
@azhavoro

Description

@azhavoro

How to reproduce:
build docker-compose -f docker-compose.yml -f components/tf_annotation/docker-compose.tf_annotation.yml -f components/cuda/docker-compose.cuda.yml up -d --build and run tf_annotation. Inference will be runned on CPU.

Seems the tensorflow_gpu package is overridden by tensorflow from datumaro requirements.

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions

    0