diff --git a/config/clusters/victor/staging.values.yaml b/config/clusters/victor/staging.values.yaml index c0e824090a..2843b11c70 100644 --- a/config/clusters/victor/staging.values.yaml +++ b/config/clusters/victor/staging.values.yaml @@ -47,6 +47,23 @@ basehub: mem_guarantee: 6.5G node_selector: node.kubernetes.io/instance-type: m5.large + profile_options: &profile_options + image: + display_name: Image + choices: + victor-notebook: + display_name: Victor Notebook + default: true + kubespawner_override: + image: quay.io/volcanocyber/victor-notebook:a045ad3616d1 + pytorch-notebook: + display_name: Pangeo ML Notebook (Pytorch) + kubespawner_override: + image: "quay.io/pangeo/pytorch-notebook:2024.09.11" + ml-notebook: + display_name: Pangeo ML Notebook (Tensorflow) + kubespawner_override: + image: "quay.io/pangeo/ml-notebook:2024.09.11" - display_name: "Medium: m5.xlarge" description: "~4 CPU, ~15G RAM" kubespawner_override: @@ -54,6 +71,7 @@ basehub: mem_guarantee: 12G node_selector: node.kubernetes.io/instance-type: m5.xlarge + profile_options: *profile_options - display_name: "Large: m5.2xlarge" description: "~8 CPU, ~30G RAM" kubespawner_override: @@ -61,6 +79,7 @@ basehub: mem_guarantee: 25G node_selector: node.kubernetes.io/instance-type: m5.2xlarge + profile_options: *profile_options - display_name: "Huge: m5.8xlarge" description: "~16 CPU, ~60G RAM" kubespawner_override: @@ -68,3 +87,17 @@ basehub: mem_guarantee: 50G node_selector: node.kubernetes.io/instance-type: m5.8xlarge + profile_options: *profile_options + - display_name: NVIDIA Tesla T4, ~16 GB, ~4 CPUs + description: "Start a container on a dedicated node with a GPU" + slug: "gpu" + kubespawner_override: + environment: + NVIDIA_DRIVER_CAPABILITIES: compute,utility + mem_limit: null + mem_guarantee: 14G + node_selector: + node.kubernetes.io/instance-type: g4dn.xlarge + extra_resource_limits: + nvidia.com/gpu: "1" + profile_options: *profile_options \ No newline at end of file