From 4dd2701ae9b25491ba7249c3e9d332a4a0c4c8d3 Mon Sep 17 00:00:00 2001
From: Gabriele Santomaggio <g.santomaggio@gmail.com>
Date: Thu, 1 Oct 2020 19:17:46 +0200
Subject: [PATCH 1/2] initial GPU SUpport

---
 install | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/install b/install
index d70f9b2..112f71c 100755
--- a/install
+++ b/install
@@ -72,6 +72,11 @@ for i in "${waiting_pod_array[@]}"; do
 done
 
 
+#### GPU Support
+kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.7.0/nvidia-device-plugin.yml
+
+
+
 info "Kubeflow pipelines ready!!"
 
 info "Defining the ingress"

From 2fbb7206a4b4a72253c523108bfbf0869ad97cad Mon Sep 17 00:00:00 2001
From: Gabriele Santomaggio <g.santomaggio@gmail.com>
Date: Mon, 5 Oct 2020 16:34:06 +0200
Subject: [PATCH 2/2] nvidia

---
 install | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/install b/install
index 112f71c..319d0d3 100755
--- a/install
+++ b/install
@@ -42,8 +42,16 @@ while [ : ]
 }
 
 
-
-curl -sfL https://get.k3s.io | sh -s - --write-kubeconfig-mode 644
+if [ $1 = "--gpu" ]
+  then   
+    info "Install Pipelines with GPU support, docker as runtime"
+    curl -sfL https://get.k3s.io | sh -s - --docker --kubelet-arg="feature-gates=DevicePlugins=true" --write-kubeconfig-mode 644
+    #### GPU Support
+    kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.7.0/nvidia-device-plugin.yml
+  else  
+    info "Install Pipelines without GPU support, containerd as runtime"
+    curl -sfL https://get.k3s.io | sh -s - --write-kubeconfig-mode 644
+fi
 
 export KUBECONFIG=/etc/rancher/k3s/k3s.yaml
 
@@ -72,10 +80,6 @@ for i in "${waiting_pod_array[@]}"; do
 done
 
 
-#### GPU Support
-kubectl create -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/v0.7.0/nvidia-device-plugin.yml
-
-
 
 info "Kubeflow pipelines ready!!"