You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
After you install tensorflow-gpu you can use the following command within the Python script that contains RetinaFace calls to:
os.environ["CUDA_VISIBLE_DEVICES"] = "1" - > Specify which GPU to use
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true' - > Limit the amount of reserved VRAM so that other scripts can be run in the same GPU.
Maybe a section in the Readme.md providing example uses of the above commands in user-created scripts could be useful as many others seem to be looking into these issues.
How can we run retina face on multiple GPU?
The text was updated successfully, but these errors were encountered: