diff --git a/ultralytics/utils/__init__.py b/ultralytics/utils/__init__.py index 3a270624..77745014 100644 --- 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: