GPU not detected in Jupyternotebook on Kubernetes GPU enabled cluster

I’m posting my solution here in the hope that it will help someone else.

In addition to modifying the singleuser: attribute as in the above post. I needed to add the following:

hub:
  revisionHistoryLimit:
  config:
    KubeSpawner:
      extra_pod_config:
        runtimeClassName: nvidia

IMO, it would be GREAT if the above config/KubeSpawner ref is added to the zero 2 hero doc's section related to deploying and using JupyterHub on a GPU enabled k8's intstance
2 Likes