From 0196d92d8ad15c36e7d67da2baafe1e663602fac Mon Sep 17 00:00:00 2001
From: buxiaomo <95112082@qq.com>
Date: Tue, 21 Jan 2025 08:56:13 +0800
Subject: [PATCH] add proxy for libnvidia-container step

---
 roles/crio/tasks/Debian.yml | 3 ++-
 roles/crio/tasks/RedHat.yml | 4 +++-
 roles/crio/tasks/SUSE.yml   | 4 +++-
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/roles/crio/tasks/Debian.yml b/roles/crio/tasks/Debian.yml
index 409213f51..d24a59475 100644
--- a/roles/crio/tasks/Debian.yml
+++ b/roles/crio/tasks/Debian.yml
@@ -20,4 +20,5 @@
   apt:
     name: "nvidia-container-runtime"
     state: present
-    update_cache: yes
\ No newline at end of file
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
\ No newline at end of file
diff --git a/roles/crio/tasks/RedHat.yml b/roles/crio/tasks/RedHat.yml
index 227f26222..8bf2d85e4 100644
--- a/roles/crio/tasks/RedHat.yml
+++ b/roles/crio/tasks/RedHat.yml
@@ -2,8 +2,10 @@
   get_url:
     url: https://nvidia.github.io/nvidia-container-runtime/{{ ansible_distribution | lower }}{{ ansible_distribution_major_version }}/nvidia-container-runtime.repo
     dest: /etc/yum.repos.d/nvidia-container-runtime.repo
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
 
 - name: Install nvidia-container-runtime
   yum:
     name: nvidia-container-runtime
-    state: latest
\ No newline at end of file
+    state: latest
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
\ No newline at end of file
diff --git a/roles/crio/tasks/SUSE.yml b/roles/crio/tasks/SUSE.yml
index b8d1630f3..b790c545a 100644
--- a/roles/crio/tasks/SUSE.yml
+++ b/roles/crio/tasks/SUSE.yml
@@ -2,6 +2,7 @@
   ansible.builtin.rpm_key:
     state: present
     key: https://nvidia.github.io/libnvidia-container/gpgkey
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
 
 - name: Add specified repository into sources
   community.general.zypper_repository:
@@ -12,4 +13,5 @@
   community.general.zypper:
     name: nvidia-container-runtime
     state: latest
-    update_cache: yes
\ No newline at end of file
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
\ No newline at end of file