From 5c8d57720c2f847f1c899ac65ed9d2f4f799e064 Mon Sep 17 00:00:00 2001
From: Nicolas Hug <nicolashug@fb.com>
Date: Thu, 15 Apr 2021 09:12:45 +0100
Subject: [PATCH] only sync cuda ifn cuda available

---
 references/detection/engine.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/references/detection/engine.py b/references/detection/engine.py
index 9f34336b0cc..49992af60a9 100644
--- a/references/detection/engine.py
+++ b/references/detection/engine.py
@@ -84,7 +84,8 @@ def evaluate(model, data_loader, device):
     for images, targets in metric_logger.log_every(data_loader, 100, header):
         images = list(img.to(device) for img in images)
 
-        torch.cuda.synchronize()
+        if torch.cuda.is_available():
+            torch.cuda.synchronize()
         model_time = time.time()
         outputs = model(images)