From 6db30d7620af868c828ae33f9f407596c7f26d4c Mon Sep 17 00:00:00 2001
From: YundongYe <8675883+ydye@users.noreply.github.com>
Date: Thu, 28 Nov 2019 10:29:30 +0800
Subject: [PATCH] [Nvidia] Upgrade GPU drivers version and new playbook to
 remove drivers (#3899)

---
 .../clean-nvidia-drivers-installed-by-apt.yml | 49 +++++++++++++++++++
 contrib/kubespray/nvidia-drivers.yml          |  4 +-
 2 files changed, 51 insertions(+), 2 deletions(-)
 create mode 100644 contrib/kubespray/clean-nvidia-drivers-installed-by-apt.yml

diff --git a/contrib/kubespray/clean-nvidia-drivers-installed-by-apt.yml b/contrib/kubespray/clean-nvidia-drivers-installed-by-apt.yml
new file mode 100644
index 0000000000..d3b055b87d
--- /dev/null
+++ b/contrib/kubespray/clean-nvidia-drivers-installed-by-apt.yml
@@ -0,0 +1,49 @@
+---
+- hosts: all
+  tasks:
+    - name: uninstall nvidia-drivers
+      apt:
+        name: nvidia-418
+        state: absent
+
+    - name: uninstall nvidia-opencl-icd
+      apt:
+        name: nvidia-opencl-icd-410
+        state: absent
+
+    - name: uninstall nvidia-prime
+      apt:
+        name: nvidia-prime
+        state: absent
+
+    - name: uninstall nvidia-settings
+      apt:
+        name: nvidia-settings
+        state: absent
+
+    - name: uninstall libcuda1
+      apt:
+        name: libcuda1-418
+        state: absent
+
+    - name: Remove dependencies that are no longer required
+      apt:
+        autoremove: yes
+
+    - name: Remove useless packages from the cache
+      apt:
+        autoclean: yes
+
+    - name: Remove the drivers repository from Ubuntu
+      apt_repository:
+        repo: ppa:graphics-drivers/ppa
+        state: absent
+
+    - name: Run the equivalent of "apt-get update" as a separate step
+      apt:
+        update_cache: yes
+
+    - name: reboot vm
+      reboot:
+
+
diff --git a/contrib/kubespray/nvidia-drivers.yml b/contrib/kubespray/nvidia-drivers.yml
index bf0e3be132..c8c10a087f 100644
--- a/contrib/kubespray/nvidia-drivers.yml
+++ b/contrib/kubespray/nvidia-drivers.yml
@@ -10,9 +10,9 @@
       apt:
         update_cache: yes
 
-    - name: Install nvidia drivers, we will install 410 version
+    - name: Install nvidia drivers, we will install 418 version
       apt:
-        name: nvidia-410
+        name: nvidia-418
         state: present
 
     - name: reboot vm