serengil / retinaface

RetinaFace: Deep Face Detection Library for Python
https://www.youtube.com/watch?v=Wm1DucuQk70&list=PLsS_1RYmYQQFdWqxQggXHynP1rqaYXv_E&index=3
MIT License
1.15k stars 150 forks source link

Which library versions do you suggest for maximum GPU performance on Windows? #70

Closed FurkanGozukara closed 9 months ago

FurkanGozukara commented 10 months ago

Dear @serengil, I am using below. Working decent. Using almost 24 GB VRAM on RTX 3090 TI

Windows 10 Python 3.10.11

Do you suggest anything else?

Installing on a fresh venv

pip3 install torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

pip install tensorflow==2.10

pip install transformers

pip install Pillow

pip install retina-face

I also compared RTX 3090 TI and RTX 3060 on same PC

Both speed are same. GPU utilization is pretty low according to nvidia-smi

image

serengil commented 9 months ago

if you install tensorflow-gpu before installing retinaface, then it will use gpu by default.