diff --git a/src/gpuhunt/_internal/constraints.py b/src/gpuhunt/_internal/constraints.py index 0f5226a..e4199f8 100644 --- a/src/gpuhunt/_internal/constraints.py +++ b/src/gpuhunt/_internal/constraints.py @@ -178,6 +178,7 @@ def is_nvidia_superchip(gpu_name: str) -> bool: NvidiaGPUInfo(name="H200NVL", memory=141, compute_capability=(9, 0)), NvidiaGPUInfo(name="B200", memory=180, compute_capability=(10, 0)), NvidiaGPUInfo(name="GB200", memory=186, compute_capability=(10, 0)), + NvidiaGPUInfo(name="B300", memory=270, compute_capability=(10, 0)), NvidiaGPUInfo(name="L4", memory=24, compute_capability=(8, 9)), NvidiaGPUInfo(name="L40", memory=48, compute_capability=(8, 9)), NvidiaGPUInfo(name="L40S", memory=48, compute_capability=(8, 9)), diff --git a/src/gpuhunt/providers/aws.py b/src/gpuhunt/providers/aws.py index 80294f0..5d3725f 100644 --- a/src/gpuhunt/providers/aws.py +++ b/src/gpuhunt/providers/aws.py @@ -378,6 +378,8 @@ def filter(cls, offers: list[RawCatalogItem]) -> list[RawCatalogItem]: "t2.small", "c5.", "m5.", + "p6-b300.", + "p6-b200.", "p5.", "p5e.", "p4d.",