From 4d58facc36bb1821783cbd9c1b5c45c4039be673 Mon Sep 17 00:00:00 2001
From: Pablo Escobar <pescobar001@gmail.com>
Date: Tue, 26 Oct 2021 19:14:54 +0200
Subject: [PATCH] add var nvidia_driver_add_repos to disable adding upstream
 repos

Signed-off-by: Pablo Escobar <pescobar001@gmail.com>
---
 defaults/main.yml                  | 1 +
 tasks/install-redhat.yml           | 5 ++++-
 tasks/install-ubuntu-cuda-repo.yml | 3 +++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/defaults/main.yml b/defaults/main.yml
index 8d9d61e..d6764de 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -4,6 +4,7 @@ nvidia_driver_persistence_mode_on: yes
 nvidia_driver_skip_reboot: no
 nvidia_driver_module_file: /etc/modprobe.d/nvidia.conf
 nvidia_driver_module_params: ''
+nvidia_driver_add_repos: yes
 
 
 ##############################################################################
diff --git a/tasks/install-redhat.yml b/tasks/install-redhat.yml
index 16112d4..d72f37b 100644
--- a/tasks/install-redhat.yml
+++ b/tasks/install-redhat.yml
@@ -32,14 +32,16 @@
   rpm_key:
     key: "{{ epel_repo_key }}"
     state: present
+  when: nvidia_driver_add_repos | bool
 
 - name: add epel repo
   become: true
   yum:
-    name: 
+    name:
       - "{{ epel_package }}"
     state: latest
   environment: "{{proxy_env if proxy_env is defined else {}}}"
+  when: nvidia_driver_add_repos | bool
 
 - name: install dependencies
   yum: name=dkms
@@ -51,6 +53,7 @@
     baseurl: "{{ nvidia_driver_rhel_cuda_repo_baseurl }}"
     gpgkey: "{{ nvidia_driver_rhel_cuda_repo_gpgkey }}"
   environment: "{{proxy_env if proxy_env is defined else {}}}"
+  when: nvidia_driver_add_repos | bool
 
 - name: install driver packages RHEL/CentOS 7 and older
   yum:
diff --git a/tasks/install-ubuntu-cuda-repo.yml b/tasks/install-ubuntu-cuda-repo.yml
index 7aab994..886b3c1 100644
--- a/tasks/install-ubuntu-cuda-repo.yml
+++ b/tasks/install-ubuntu-cuda-repo.yml
@@ -11,12 +11,14 @@
     owner: "root"
     group: "root"
     mode: "0644"
+  when: nvidia_driver_add_repos | bool
 
 - name: add key
   apt_key:
     url: "{{ nvidia_driver_ubuntu_cuda_repo_gpgkey_url }}"
     id: "{{ nvidia_driver_ubuntu_cuda_repo_gpgkey_id }}"
   environment: "{{proxy_env if proxy_env is defined else {}}}"
+  when: nvidia_driver_add_repos | bool
 
 
 - name: add repo
@@ -24,6 +26,7 @@
     repo: "deb {{ nvidia_driver_ubuntu_cuda_repo_baseurl }} /"
     update_cache: yes
   environment: "{{proxy_env if proxy_env is defined else {}}}"
+  when: nvidia_driver_add_repos | bool
 
 - name: install driver packages
   apt: