GazeTracking icon indicating copy to clipboard operation
GazeTracking copied to clipboard

Published 20 hours ago • verified publisher icon antoinelame

Running on gpu

Open gd1925 opened this issue 2 years ago • 3 comments

Hi, Thank you for the awesome work! It is very helpful! Is it possible to run the code on GPU? I am running it on Ubuntu 18.04 and when I try to run, it does not run on gpu devices and runs on cpu. Is there any way to enforce the condition to run the example.py on gpu? I also tried to run export CUDA_VISIBLE_DEVICES=0, but it still seems to be running on cpu. Many thanks!

gd1925 avatar Dec 21 '22 10:12 gd1925

To my knowledge, it runs on CPU only no matter what.

Cuda-Chen avatar Dec 22 '22 06:12 Cuda-Chen

Hi @Cuda-Chen, do you think trying to run the dlib library on gpu can help?

gd1925 avatar Dec 23 '22 05:12 gd1925

@gd1925 It is possible to let dlib and OpenCV function run on GPU. However, the bus to GPU will be the bottleneck. Unless you have a particular reason to run on GPU (e.g., given a 4K resolution image and find the face), running on GPU will not be an idea.

Cuda-Chen avatar Dec 23 '22 10:12 Cuda-Chen