From c8388d952f5559cf0ca217fc47836b2a9ad88414 Mon Sep 17 00:00:00 2001 From: Lakshantha Dissanayake Date: Mon, 8 Apr 2024 12:52:19 -0700 Subject: [PATCH] Fix proc_device_model name for NVIDIA Jetson (#9876) Co-authored-by: Glenn Jocher --- ultralytics/utils/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: