force nvidia

oceanprotocol · Dec 13, 2023 · 7cb9ceb · 7cb9ceb
1 parent 5d562ca
commit 7cb9ceb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/operator_engine/resources.py b/operator_engine/resources.py
@@ -324,6 +324,7 @@ def create_algorithm_job(body, logger, resources):
     job["spec"]["template"]["spec"]["containers"][0]["resources"]["requests"][
         "cpu"
     ] = resources["requests_cpu"]
+    job["spec"]["template"]["spec"]["containers"][0]["resources"]["requests"]["nvidia.com/gpu"]="1"
     job["spec"]["template"]["spec"]["containers"][0]["resources"]["limits"] = dict()
     job["spec"]["template"]["spec"]["containers"][0]["resources"]["limits"][
         "memory"