Docker support for CUDA environment

vietmoney / face-anti-spoofing

A small application that enable to detect fake or retake face photo

Apache License 2.0

17 stars 7 forks source link

Docker support for CUDA environment #5

Open vndee opened 3 years ago

vndee commented 3 years ago

Hi, thanks for your great work. How can I find a version of docker-compose that support CUDA environment for this project? And does it support multi-gpu inference to speed up processing time?

TinDang97 commented 3 years ago

Hi @vndee. Currently, DockerFile just using CPU to process. I will support Nvidia Docker next few days.

Thanks.