Fix proc_device_model name for NVIDIA Jetson (#9876)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
7 months ago · c8388d952f
parent f2ddc96334
commit c8388d952f
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/utils/init.py
+++ b/ultralytics/utils/init.py
@ -498,7 +498,7 @@ def is_jetson() -> bool:
    Returns:
        (bool): True if running on a Jetson Nano or Jetson Orin, False otherwise.
    """
-    return "Jetson" in PROC_DEVICE_MODEL  # i.e. "Jetson Nano" or "Jetson Orin"
+    return "NVIDIA" in PROC_DEVICE_MODEL  # i.e. "NVIDIA Jetson Nano" or "NVIDIA Orin NX"


 def is_online() -> bool: