search

saimj7 / People-Counting-in-Real-Time

People Counting in Real-Time with an IP camera.
MIT License
528 stars 263 forks source link

launch using CUDA #71

Closed isakimov closed 11 months ago

isakimov commented 1 year ago

Hello! I'm trying to run a counter with CUDA to take the load off the CPU Installed all the necessary dependencies (CUDA, CuDNN, torch, torchvision, torchaudio), torch.cuda.is_available() returns True, but when people-counter.py is run, the process is not displayed in nvidia-smi. In gpustat I see that all the load goes to the CPU. Has anyone managed to do this?

jasonjiang8866 commented 1 year ago

you need to compile opencv against CUDA