bump gpu resources

latchbio · Dec 6, 2024 · 0809417 · 0809417
1 parent 08b0197
commit 0809417
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/latch/resources/tasks.py b/src/latch/resources/tasks.py
@@ -56,13 +56,13 @@ def get_v100_x1_pod() -> Pod:
             "cpu": "7",
             "memory": "48Gi",
             "nvidia.com/gpu": 1,
-            "ephemeral-storage": "1500Gi",
+            "ephemeral-storage": "4500Gi",
         },
         limits={
             "cpu": "7",
             "memory": "48Gi",
             "nvidia.com/gpu": 1,
-            "ephemeral-storage": "2000Gi",
+            "ephemeral-storage": "5000Gi",
         },
     )
     primary_container.resources = resources
@@ -92,13 +92,13 @@ def get_v100_x4_pod() -> Pod:
             "cpu": "30",
             "memory": "230Gi",
             "nvidia.com/gpu": 4,
-            "ephemeral-storage": "1500Gi",
+            "ephemeral-storage": "4500Gi",
         },
         limits={
             "cpu": "30",
             "memory": "230Gi",
             "nvidia.com/gpu": 4,
-            "ephemeral-storage": "2000Gi",
+            "ephemeral-storage": "5000Gi",
         },
     )
     primary_container.resources = resources
@@ -133,13 +133,13 @@ def get_v100_x8_pod() -> Pod:
             "cpu": "62",
             "memory": "400Gi",
             "nvidia.com/gpu": 8,
-            "ephemeral-storage": "1500Gi",
+            "ephemeral-storage": "4500Gi",
         },
         limits={
             "cpu": "62",
             "memory": "400Gi",
             "nvidia.com/gpu": 8,
-            "ephemeral-storage": "2000Gi",
+            "ephemeral-storage": "5000Gi",
         },
     )
     primary_container.resources = resources