backport nvidia device location change fix (#74)

Co-authored-by: Valentin Rutz <val.rutz@dataiku.com>

Author: thtrunck

Changelog.md

@@ -1,5 +1,8 @@
 # Changelog
 
+## Version 2.1.1 - Bugfix release
+- Update Nvidia driver location
+
 ## Version 2.1.0 - Feature release
 
 - Support GPU out-of-the-box

plugin.json

@@ -1,6 +1,6 @@
 {
     "id": "aks-clusters",
-    "version": "2.1.0",
+    "version": "2.1.1",
     "meta": {
         "label": "AKS clusters",
         "description": "Interact with or create Microsoft Azure Kubernetes Service clusters",

python-lib/dku_kube/nvidia_utils.py

@@ -12,7 +12,7 @@ def has_gpu_driver(kube_config_path):
 
 def add_gpu_driver_if_needed(kube_config_path):
     if not has_gpu_driver(kube_config_path):
-        cmd = ['kubectl', 'apply', '-f', "https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/master/nvidia-device-plugin.yml"]
+        cmd = ['kubectl', 'apply', '-f', "https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/main/deployments/static/nvidia-device-plugin.yml"]
 
         logging.info("Install NVIDIA GPU drivers with : %s" % json.dumps(cmd))
         env = os.environ.copy()