From 2b7c0a708e0c370b631e77db8d7ca3211cb0a47b Mon Sep 17 00:00:00 2001
From: aku-ato <111570992+aku-ato@users.noreply.github.com>
Date: Tue, 26 Mar 2024 15:34:03 +0100
Subject: [PATCH] Conditional use of GPU on FastMtCnn

---
 deepface/detectors/FastMtCnn.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deepface/detectors/FastMtCnn.py b/deepface/detectors/FastMtCnn.py
index 2df5c4d..c6a28a0 100644
--- a/deepface/detectors/FastMtCnn.py
+++ b/deepface/detectors/FastMtCnn.py
@@ -68,7 +68,9 @@ class FastMtCnnClient(Detector):
                 "Please install using 'pip install facenet-pytorch' "
             ) from e
 
-        face_detector = fast_mtcnn(device="cuda:0")
+        device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+        face_detector = fast_mtcnn(device=device)
+
         return face_detector