Importing torch on yolov5 webcam demo to fix cuDNN error

opendr-eu · Sep 25, 2023 · 5046ec9 · 5046ec9
1 parent 30817fe
commit 5046ec9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py b/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py
@@ -16,6 +16,8 @@
 
 import cv2
 import time
+# Importing torch to avoid "RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED" error
+import torch  # NOQA
 
 from opendr.engine.data import Image
 from opendr.perception.object_detection_2d import YOLOv5DetectorLearner