matrice-compute 0.1.29__py3-none-any.whl → 0.1.30__py3-none-any.whl

matrice_compute/scaling.py

@@ -1266,6 +1266,38 @@ class Scaling:
                 if "jetson" in model or "tegra" in model:
                     is_jetson = True
                     gpu_provider = "NVIDIA"
+                    
+                    # Detect specific Jetson model for GPU architecture
+                    if "orin" in model:
+                        if "agx" in model:
+                            gpu_arch_family = "Jetson Orin AGX"
+                        elif "nx" in model:
+                            gpu_arch_family = "Jetson Orin NX"
+                        elif "nano" in model:
+                            gpu_arch_family = "Jetson Orin Nano"
+                        else:
+                            gpu_arch_family = "Jetson Orin"
+                    elif "thor" in model:
+                        gpu_arch_family = "Jetson Thor"
+                    elif "xavier" in model:
+                        if "agx" in model:
+                            gpu_arch_family = "Jetson Xavier AGX"
+                        elif "nx" in model:
+                            gpu_arch_family = "Jetson Xavier NX"
+                        else:
+                            gpu_arch_family = "Jetson Xavier"
+                    elif "nano" in model and "orin" not in model:
+                        gpu_arch_family = "Jetson Nano"
+                    elif "tx2" in model:
+                        gpu_arch_family = "Jetson TX2"
+                    elif "tx1" in model:
+                        gpu_arch_family = "Jetson TX1"
+                    else:
+                        gpu_arch_family = "Jetson (Unknown Model)"
+                    
+                    # Set gpu_arch to the full model string for detailed info
+                    gpu_arch = model.strip()
+                    
                     try:
                         cuda_result = subprocess.run(["nvcc", "--version"], capture_output=True, text=True)
                         if cuda_result.returncode == 0:
@@ -1275,6 +1307,15 @@ class Scaling:
                                     break
                     except Exception:
                         pass
+                    # Fallback to nvidia-smi for CUDA version if nvcc failed
+                    if not cuda_version:
+                        try:
+                            nvidia_smi_result = subprocess.run(["nvidia-smi", "--query-gpu=driver_version", "--format=csv,noheader"], capture_output=True, text=True)
+                            if nvidia_smi_result.returncode == 0:
+                                # nvidia-smi doesn't directly give CUDA version, but we can infer it's available
+                                cuda_version = "Available (via nvidia-smi)"
+                        except Exception:
+                            pass
         except Exception:
             pass
 

{matrice_compute-0.1.29.dist-info → matrice_compute-0.1.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_compute
-Version: 0.1.29
+Version: 0.1.30
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_compute-0.1.29.dist-info → matrice_compute-0.1.30.dist-info}/RECORD

@@ -8,11 +8,11 @@ matrice_compute/instance_utils.py,sha256=N4yPDvNukFEEBngR0lEt4x_XT5hur1q0P-spM2x
 matrice_compute/prechecks.py,sha256=W9YmNF3RcLhOf4U8WBlExvFqDw1aGWSNTlJtA73lbDQ,17196
 matrice_compute/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matrice_compute/resources_tracker.py,sha256=1jSLrIFlOh-vgyNzFrUrE2Ak2JAGCIfV7wcyEPJ0f2c,32246
-matrice_compute/scaling.py,sha256=oeCgQb6g9W7p8-VaDXT4jDH3aMz-fCO5MZmj-1Gbz1c,52931
+matrice_compute/scaling.py,sha256=cdEJqdVsPGDeOjkVAG85lubOn-qwDRV5qqmrNl_XpCM,55146
 matrice_compute/shutdown_manager.py,sha256=0MYV_AqygqR9NEntYf7atUC-PbWXyNkm1f-8c2aizgA,13234
 matrice_compute/task_utils.py,sha256=3qIutiQdYPyGRxH9ZwLbqdg8sZcnp6jp08pszWCRFl0,2820
-matrice_compute-0.1.29.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_compute-0.1.29.dist-info/METADATA,sha256=5NCF_K1dBcVouNXdfnUL7cFm7fg7PojuURK2m1if9-Y,1038
-matrice_compute-0.1.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_compute-0.1.29.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
-matrice_compute-0.1.29.dist-info/RECORD,,
+matrice_compute-0.1.30.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_compute-0.1.30.dist-info/METADATA,sha256=kTDi915lwYwQJWMEyAT1F5GT5GfUW2SQfon5ki_X9tM,1038
+matrice_compute-0.1.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_compute-0.1.30.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
+matrice_compute-0.1.30.dist-info/RECORD,,