<p>I installed the <code>Dask Helm Chart</code> with a revised <code>values.yaml</code> to have <code>10 workers</code>, however instead of <code>Dask Workers</code> I want to create <code>Dash CUDA Workers</code> to take advantage of the NVIDIA GPUs on my <code>multi-node</code> cluster.</p>
<p>I tried to modify the <code>values.yaml</code> as follows to get <code>Dask CUDA workers</code> instead of <code>Dask Workers</code>, but the worker pods are able to start. I did already install the NVIDIA GPUs on all my nodes on the Kubernetes per the official instructions so I'm not sure what DASK needs to see in order to create the Dask-Cuda-Workers.</p>
<pre><code>worker:
  name: worker
  image:
    repository: &quot;daskdev/dask&quot;
    tag: 2.19.0
    pullPolicy: IfNotPresent
    dask_worker: &quot;dask-cuda-worker&quot;
    #dask_worker: &quot;dask-worker&quot;
    pullSecrets:
    #  - name: regcred
  replicas: 15
  default_resources:  # overwritten by resource limits if they exist
    cpu: 1
    memory: &quot;4GiB&quot;
  env:
  #  - name: EXTRA_APT_PACKAGES
  #    value: build-essential openssl
  #  - name: EXTRA_CONDA_PACKAGES
  #    value: numba xarray -c conda-forge
  #  - name: EXTRA_PIP_PACKAGES
  #    value: s3fs dask-ml --upgrade
  resources: {}
  #  limits:
  #    cpu: 1
  #    memory: 3G
  #    nvidia.com/gpu: 1
  #  requests:
  #    cpu: 1
  #    memory: 3G
  #    nvidia.com/gpu: 1
</code></pre>

Vinh Tran
<p>As dask-cuda-worker is not yet officially in the dask image you will need to pull the image a different image: <code>rapidsai/rapidsai:latest</code></p>
