add proxy for libnvidia-container step

buxiaomo · Jan 21, 2025 · 42a9b7a · 42a9b7a
1 parent 9418257
commit 42a9b7a
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 6 deletions.
diff --git a/roles/containerd/tasks/Debian.yml b/roles/containerd/tasks/Debian.yml
@@ -20,4 +20,5 @@
   apt:
     name: "nvidia-container-runtime"
     state: present
-    update_cache: yes
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
diff --git a/roles/containerd/tasks/RedHat.yml b/roles/containerd/tasks/RedHat.yml
@@ -2,8 +2,11 @@
   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
+    state: latest
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
diff --git a/roles/containerd/tasks/SUSE.yml b/roles/containerd/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
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
diff --git a/roles/docker/tasks/Debian.yml b/roles/docker/tasks/Debian.yml
@@ -20,4 +20,5 @@
   apt:
     name: "nvidia-container-runtime"
     state: present
-    update_cache: yes
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
diff --git a/roles/docker/tasks/RedHat.yml b/roles/docker/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
+    state: latest
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"
diff --git a/roles/docker/tasks/SUSE.yml b/roles/docker/tasks/SUSE.yml
@@ -2,6 +2,8 @@
   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 +14,5 @@
   community.general.zypper:
     name: nvidia-container-runtime
     state: latest
-    update_cache: yes
+    update_cache: yes
+  environment: "{{ proxy_env | default({'http_proxy': '', 'https_proxy': '', 'no_proxy': ''}) }}"