able-ice-99180
07/29/2022, 12:39 PMplugins:
k8s:
resource-tolerations:
- <http://nvidia.com/gpu:|nvidia.com/gpu:>
- key: "key1"
operator: "Equal"
value: "value1"
effect: "NoSchedule"
I saw this thread which mentioned updating the propellor configmap. Is that the way to go? Or is there a way to update plugins via the flyte config?
Thanks!thankful-minister-83577
able-ice-99180
07/29/2022, 5:05 PMthankful-minister-83577
able-ice-99180
07/29/2022, 5:07 PMable-ice-99180
07/29/2022, 5:10 PMthankful-minister-83577
thankful-minister-83577
able-ice-99180
07/29/2022, 6:05 PMgifted-house-14547
08/09/2022, 5:49 AMcontainer = V1Container(
name="container",
resources=V1ResourceRequirements(
requests={"cpu": "1", "memory":"4Gi", "<http://nvidia.com|nvidia.com>":"1"},
limits={"cpu": "4", "memory":"16Gi", "<http://nvidia.com|nvidia.com>":"1"},
) ,
image=<your image name>,
)
pod_spec = V1PodSpec(
containers=[container],
image_pull_secrets=[V1LocalObjectReference(<your image_account>)],
volumes=volumes,
tolerations=<your node toleration which want to assign resource>,
node_selector=<you will need some specific node selector when there are the same spec nodes>
)
@task(
task_config=Pod(
pod_spec=pod_spec,
),
)
def your_task()->None:
print("this is test")