Update object_classification_yolov8.py

uug-ai · Nov 27, 2024 · 6f002bd · 6f002bd
1 parent df9d0c7
commit 6f002bd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/object_classification_yolov8.py b/object_classification_yolov8.py
@@ -23,6 +23,10 @@
 from uugai_python_dynamic_queue.MessageBrokers import RabbitMQ
 from uugai_python_kerberos_vault.KerberosVault import KerberosVault
 
+# Following error is thrown: [W NNPACK.cpp:64] Could not initialize NNPACK! Reason: Unsupported hardware.
+# https://stackoverflow.com/questions/69711410/could-not-initialize-nnpack
+torch.backends.nnpack.enabled = False
+
 # Initialize the VariableClass object, which contains all the necessary environment variables.
 var = VariableClass()