lyceum-cli 1.0.29__py3-none-any.whl → 1.0.31__py3-none-any.whl

lyceum/external/compute/execution/gpu_selection.py

@@ -44,6 +44,21 @@ GPU_DISPLAY_NAMES = {
     "gpu.rtx6000pro": "RTX 6000 Pro",
 }
 
+# VRAM in GB for each GPU profile (for showing excluded GPUs)
+GPU_VRAM_GB = {
+    "gpu": 16,
+    "gpu.t4": 16,
+    "gpu.t4.64gb": 16,
+    "gpu.a100": 40,
+    "gpu.a100.40gb": 40,
+    "gpu.a100.80gb": 80,
+    "gpu.h100": 80,
+    "gpu.h200": 141,
+    "gpu.l40s": 48,
+    "gpu.b200": 180,
+    "gpu.rtx6000pro": 48,
+}
+
 
 def format_gpu_name(profile: str) -> str:
     """Format GPU profile name for display."""
@@ -342,6 +357,7 @@ def display_results(data: dict, file_path: str | None = None) -> None:
 
         # Sort by VRAM size for better display
         sorted_configs = sorted(minimal_configs, key=lambda x: x.get("per_gpu_vram_gb", 0))
+        compatible_gpu_types = {cfg.get("gpu_type") for cfg in minimal_configs}
 
         for i, cfg in enumerate(sorted_configs):
             gpu_type = format_gpu_name(cfg.get("gpu_type", "?"))
@@ -362,6 +378,42 @@ def display_results(data: dict, file_path: str | None = None) -> None:
 
         console.print(gpu_table)
 
+        # Show excluded GPUs (those not in minimal_configs)
+        # Get total memory required from extraction result
+        mem_reqs = mem_config.get("memory_requirements", {})
+        total_mem_gb = 0
+        if mem_reqs:
+            # Sum up memory components
+            total_mem_gb = (
+                mem_reqs.get("model_weights", 0) +
+                mem_reqs.get("gradients", 0) +
+                mem_reqs.get("optimizer_states", 0) +
+                mem_reqs.get("activations", 0)
+            )
+
+        if not total_mem_gb and sorted_configs:
+            # Estimate from minimal config utilization
+            first_cfg = sorted_configs[0]
+            vram = first_cfg.get("per_gpu_vram_gb", 0)
+            util = first_cfg.get("vram_utilization_percent", 0)
+            if vram and util:
+                total_mem_gb = vram * (util / 100)
+
+        # Find GPUs that were excluded (deduplicate by display name)
+        excluded_gpus = {}
+        for profile, vram in GPU_VRAM_GB.items():
+            if profile not in compatible_gpu_types and profile in ("gpu", "gpu.t4"):
+                gpu_name = format_gpu_name(profile)
+                # Only keep one entry per display name (e.g., "T4")
+                if gpu_name not in excluded_gpus:
+                    excluded_gpus[gpu_name] = vram
+
+        if excluded_gpus and total_mem_gb:
+            console.print()
+            console.print("[dim]Excluded GPUs (insufficient VRAM):[/dim]")
+            for gpu_name, vram in sorted(excluded_gpus.items(), key=lambda x: x[1]):
+                console.print(f"[dim]  • {gpu_name}: {vram} GB available, ~{total_mem_gb:.1f} GB required[/dim]")
+
     # Show run command hint if we have a best GPU and file path
     if best_gpu and file_path:
         machine_flag = best_gpu.get("gpu_type", "gpu")

{lyceum_cli-1.0.29.dist-info → lyceum_cli-1.0.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lyceum-cli
-Version: 1.0.29
+Version: 1.0.31
 Summary: Command-line interface for Lyceum Cloud Execution API
 Home-page: https://lyceum.technology
 Author: Lyceum Team

{lyceum_cli-1.0.29.dist-info → lyceum_cli-1.0.31.dist-info}/RECORD

@@ -8,7 +8,7 @@ lyceum/external/compute/execution/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZ
 lyceum/external/compute/execution/config.py,sha256=6JJgLJnDPTwevEaNdB1nEICih_qbBmws5u5_S9gj7k0,8866
 lyceum/external/compute/execution/docker.py,sha256=0Y6lxJAm56Jrl0HxeNz1mX6DGs556i2iMN9_U1JQP0c,9635
 lyceum/external/compute/execution/docker_compose.py,sha256=YsWPnw5nB1ZpqjU9X8o_klT78I5m46PapVwVEeWra_8,9189
-lyceum/external/compute/execution/gpu_selection.py,sha256=fzYW6k8exMbGnYpZTDKpALRrlKAO5RrppLUq6z2UWoo,37059
+lyceum/external/compute/execution/gpu_selection.py,sha256=B6x8enjdH07eUR4s1eobxGmekLyJ8-IbadxFjOAjApw,39109
 lyceum/external/compute/execution/notebook.py,sha256=Gw9UhJ-UjYhpjdIYQ4IMYhVjhSkAFpOQ9aFYj1qOeww,7542
 lyceum/external/compute/execution/python.py,sha256=8Y9ElWs9RdauQbhECKcBPSoT0XZeGhXZ_pkEpr3sGro,12878
 lyceum/external/compute/execution/workloads.py,sha256=4fsRWbYGmsQMGPPIN1jUG8cG5NPG9yV26ANJ-DtaXqc,5844
@@ -29,8 +29,8 @@ lyceum/shared/streaming.py,sha256=wFb7w7fra63y8WWaIA8_E1Z6Sx_6G-0J53Zh010eZgk,93
 lyceum_cloud_execution_api_client/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 lyceum_cloud_execution_api_client/api/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 lyceum_cloud_execution_api_client/models/__init__.py,sha256=AMlb9R9O9aNC9hvKz_8TFpEfOolYC3VtFS5JX17kYks,4888
-lyceum_cli-1.0.29.dist-info/METADATA,sha256=3ltYxfRInW369XotoAGtTzZ7j8uj5i0vPDpu6hI8rJM,1482
-lyceum_cli-1.0.29.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-lyceum_cli-1.0.29.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
-lyceum_cli-1.0.29.dist-info/top_level.txt,sha256=CR7FEMloAXgLsHUR6ti3mWNcpgje27HRHSfq8doIils,41
-lyceum_cli-1.0.29.dist-info/RECORD,,
+lyceum_cli-1.0.31.dist-info/METADATA,sha256=bGp9mqLakov3SeduGI7Fo_V5jdsHyEzh5UsisZgBhK8,1482
+lyceum_cli-1.0.31.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+lyceum_cli-1.0.31.dist-info/entry_points.txt,sha256=Oq-9wDkxVd6MHgNiUTYwXI9SGhvR3VkD7Mvk0xhiUZo,43
+lyceum_cli-1.0.31.dist-info/top_level.txt,sha256=CR7FEMloAXgLsHUR6ti3mWNcpgje27HRHSfq8doIils,41
+lyceum_cli-1.0.31.dist-info/RECORD,,