add nvidia to config/publish

oceanprotocol · Dec 13, 2023 · 018665f · 018665f
1 parent 36dfd24
commit 018665f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/operator_engine/resources.py b/operator_engine/resources.py
@@ -231,6 +231,7 @@ def create_configure_job(body, logger):
     job["spec"]["template"]["spec"]["containers"][0]["volumeMounts"].append(
         volume_mount
     )
+    job["spec"]["template"]["spec"]["containers"][0]["resources"]["limits"]["nvidia.com/gpu"]="1"
     job = jobs_common_params(job, logger)
     create_job(logger, body, job)
 
@@ -633,7 +634,7 @@ def create_publish_job(body, logger):
     job["spec"]["template"]["spec"]["containers"][0]["volumeMounts"].append(
         volume_mount
     )
-
+    job["spec"]["template"]["spec"]["containers"][0]["resources"]["limits"]["nvidia.com/gpu"]="1"
     # set the account
     job["spec"]["template"]["spec"]["serviceAccount"] = OperatorConfig.SERVICE_ACCOUNT
     job["spec"]["template"]["spec"][