#100 Set to CPU mode if no cuda support

ibaiGorordo · Apr 8, 2023 · 6660e1a · 6660e1a
1 parent 957ca8b
commit 6660e1a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pykinect_azure/k4abt/_k4abt.py b/pykinect_azure/k4abt/_k4abt.py
@@ -31,6 +31,7 @@ def setup_onnx_provider_linux():
 		ctypes.cdll.LoadLibrary("libonnxruntime_providers_cuda.so")
 	except Exception as e:
 		ctypes.cdll.LoadLibrary("libonnxruntime.so.1.10.0")
+		k4abt_tracker_default_configuration.processing_mode = K4ABT_TRACKER_PROCESSING_MODE_CPU
 
 def setup_onnx_provider_windows():
 	try: