From 018a9d06b05318ec72237c8c06537830a25ef47d Mon Sep 17 00:00:00 2001
From: YuviPanda <yuvipanda@gmail.com>
Date: Mon, 5 Aug 2024 13:45:03 -0700
Subject: [PATCH] neurohackedemy: Add dropdown to select GPU in unlisted choice

Ref https://2i2c.freshdesk.com/a/tickets/1971
---
 .../2i2c-aws-us/neurohackademy.values.yaml    | 29 +++++++++++++++----
 1 file changed, 24 insertions(+), 5 deletions(-)

diff --git a/config/clusters/2i2c-aws-us/neurohackademy.values.yaml b/config/clusters/2i2c-aws-us/neurohackademy.values.yaml
index eac6d9cb3..de6cabbaa 100644
--- a/config/clusters/2i2c-aws-us/neurohackademy.values.yaml
+++ b/config/clusters/2i2c-aws-us/neurohackademy.values.yaml
@@ -123,11 +123,7 @@ jupyterhub:
       - display_name: "Bring your own image"
         description: Specify your own docker image (must have python and jupyterhub installed in it)
         slug: custom
-        kubespawner_override:
-          cpu_guarantee: 0.5
-          cpu_limit: 14
-          mem_guarantee: 4G
-          mem_limit: 16G
+        kubespawner_override: {}
         profile_options:
           image:
             display_name: Image
@@ -140,6 +136,29 @@ jupyterhub:
                 image: "{value}"
                 init_containers: *init_containers
             choices: {}
+          gpu:
+            display_name: GPU
+            choices:
+              01-none:
+                display_name: No GPU
+                default: true
+                kubespawner_override:
+                  cpu_guarantee: 0.5
+                  cpu_limit: 14
+                  mem_guarantee: 4G
+                  mem_limit: 16G
+              01-yes:
+                display_name: Yes (NVIDIA T4)
+                kubespawner_override:
+                  mem_limit: null
+                  mem_guarantee: 14G
+                  environment:
+                    NVIDIA_DRIVER_CAPABILITIES: compute,utility
+                  node_selector:
+                    node.kubernetes.io/instance-type: g4dn.xlarge
+                    2i2c.org/community: neurohackademy
+                  extra_resource_limits:
+                    nvidia.com/gpu: "1"
   hub:
     config:
       JupyterHub: