davisking/dlib

Choose backend (GPU/CPU)

Open

#1.852 geöffnet am 31. Juli 2019

Auf GitHub ansehen
 (12 Kommentare) (1 Reaktion) (0 zugewiesene Personen)C++ (3.314 Forks)batch import
enhancementhelp wanted

Repository-Metriken

Stars
 (12.570 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

So I couldn't find a way to do this in the documentation but I was wondering if there was a way to force dlib to use the CPU if it's built with CUDA enabled? I ask because I'm making a multithreaded application with dlib and only a maximum of 2 threads at a time can use CUDA (because of limitations of the GPU itself). So I'd like to in a thread get dlib to use the GPU in the thread if <2 threads are using the GPU and otherwise fallback to using the CPU.

Also a slightly related question how would dlib cope with a machine with multiple GPUs installed all for compute purposes?

Contributor Guide