Allocate specific GPUs by Node selector (K8S)

smlee77
smlee77 Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Adv Designer, Registered Posts: 1

Hello.

I installed K8S, Dataiku locally (on-premies).

The cluster consists of one master node and one worker node which has 4 GPUs

I want to allocate specific GPUs by using 'Node selector' option in 'Containerized execution' tab.

what I wrote is here

nodeSelector:
gpu: gpu
env:
- name: NVIDIA_VISIBLE_DEVICES
value: "2,3"

* I named worker node as gpu
The errors said it failed to create container and terminated.
Anyone knows the solution?
Thanks in advance

Operating system used: ubuntu
Tagged:
Setup Info
    Tags
      Help me…