Allow customizing container image path used in NVIDIA GPU addon. (#4229)

inventory/sample/group_vars/k8s-cluster/k8s-cluster.yml

@@ -210,3 +210,8 @@ persistent_volumes_enabled: false
 # nvidia_driver_version: "384.111"
 ## flavor can be tesla or gtx
 # nvidia_gpu_flavor: gtx
+## NVIDIA driver installer images. Change them if you have trouble accessing gcr.io.
+# nvidia_driver_install_centos_container: atzedevries/nvidia-centos-driver-installer:2
+# nvidia_driver_install_ubuntu_container: gcr.io/google-containers/ubuntu-nvidia-driver-installer@sha256:eea7309dc4fa4a5c9d716157e74b90826e0a853aa26c7219db4710ddcd1ad8bc
+## NVIDIA GPU device plugin image.
+# nvidia_gpu_device_plugin_container: "k8s.gcr.io/nvidia-gpu-device-plugin@sha256:0842734032018be107fa2490c98156992911e3e1f2a21e059ff0105b07dd8e9e"

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/defaults/main.yml

@@ -4,7 +4,10 @@ nvidia_driver_version: "390.87"
 nvidia_gpu_tesla_base_url: https://us.download.nvidia.com/tesla/
 nvidia_gpu_gtx_base_url: http://us.download.nvidia.com/XFree86/Linux-x86_64/
 nvidia_gpu_flavor: tesla
-nvidia_url_end: "{{nvidia_driver_version}}/NVIDIA-Linux-x86_64-{{nvidia_driver_version}}.run"
+nvidia_url_end: "{{ nvidia_driver_version }}/NVIDIA-Linux-x86_64-{{ nvidia_driver_version }}.run"
 nvidia_driver_install_container: false
+nvidia_driver_install_centos_container: atzedevries/nvidia-centos-driver-installer:2
+nvidia_driver_install_ubuntu_container: gcr.io/google-containers/ubuntu-nvidia-driver-installer@sha256:eea7309dc4fa4a5c9d716157e74b90826e0a853aa26c7219db4710ddcd1ad8bc
 nvidia_driver_install_supported: false
+nvidia_gpu_device_plugin_container: "k8s.gcr.io/nvidia-gpu-device-plugin@sha256:0842734032018be107fa2490c98156992911e3e1f2a21e059ff0105b07dd8e9e"
 nvidia_gpu_nodes: []

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/k8s-device-plugin-nvidia-daemonset.yml.j2

@@ -38,7 +38,7 @@ spec:
         hostPath:
           path: /dev
       containers:
-      - image: "k8s.gcr.io/nvidia-gpu-device-plugin@sha256:0842734032018be107fa2490c98156992911e3e1f2a21e059ff0105b07dd8e9e"
+      - image: "{{ nvidia_gpu_device_plugin_container }}"
         command: ["/usr/bin/nvidia-gpu-device-plugin", "-logtostderr"]
         name: nvidia-gpu-device-plugin
         resources:

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/nvidia-driver-install-daemonset.yml.j2

@@ -48,7 +48,7 @@ spec:
         hostPath:
           path: /
       initContainers:
-      - image: "{{nvidia_driver_install_container}}"
+      - image: "{{ nvidia_driver_install_container }}"
         name: nvidia-driver-installer
         resources:
           requests:
@@ -63,9 +63,9 @@ spec:
           - name: ROOT_MOUNT_DIR
             value: /root
           - name: NVIDIA_DRIVER_VERSION
-            value: "{{nvidia_driver_version}}"
+            value: "{{ nvidia_driver_version }}"
           - name: NVIDIA_DRIVER_DOWNLOAD_URL
-            value: "{{nvidia_driver_download_url_default}}"
+            value: "{{ nvidia_driver_download_url_default }}"
         volumeMounts:
         - name: nvidia-install-dir-host
           mountPath: /usr/local/nvidia
@@ -74,5 +74,5 @@ spec:
         - name: root-mount
           mountPath: /root
       containers:
-      - image: "gcr.io/google-containers/pause:2.0"
+      - image: "{{ pod_infra_image_repo }}"
         name: pause

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/vars/centos-7.yml

 ---
-nvidia_driver_install_container: atzedevries/nvidia-centos-driver-installer:2
+nvidia_driver_install_container: "{{ nvidia_driver_install_centos_container }}"
 nvidia_driver_install_supported: true

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/vars/ubuntu-16.yml

 ---
-nvidia_driver_install_container: gcr.io/google-containers/ubuntu-nvidia-driver-installer@sha256:eea7309dc4fa4a5c9d716157e74b90826e0a853aa26c7219db4710ddcd1ad8bc
+nvidia_driver_install_container: "{{ nvidia_driver_install_ubuntu_container }}"
 nvidia_driver_install_supported: true

roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/vars/ubuntu-18.yml

 ---
-nvidia_driver_install_container: gcr.io/google-containers/ubuntu-nvidia-driver-installer@sha256:eea7309dc4fa4a5c9d716157e74b90826e0a853aa26c7219db4710ddcd1ad8bc
+nvidia_driver_install_container: "{{ nvidia_driver_install_ubuntu_container }}"
 nvidia_driver_install_supported: true