Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
nvidia_container_hook_repo_base_url: 'https://nvidia.github.io/'
libnvidia_container_suffix: 'libnvidia-container/centos7/$basearch'
libnvidia_container_repo_gpgcheck: true
libnvidia_container_gpgcheck: false
nvidia_container_runtime_suffix: 'nvidia-container-runtime/centos7/$basearch'
nvidia_container_runtime_repo_gpgcheck: true
nvidia_container_runtime_gpgcheck: false
27 changes: 25 additions & 2 deletions playbooks/roles/nvidia-container-runtime-hook/tasks/main.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,29 @@
---
- name: install NVIDIA repo for container runtime hook
shell: yum-config-manager --add-repo=https://nvidia.github.io/nvidia-container-runtime/centos7/x86_64/nvidia-container-runtime.repo
- name: libnvidia-container repo
yum_repository:
name: libnvidia-container
description: libnvidia-container
baseurl: '{{ nvidia_container_hook_repo_base_url }}{{ libnvidia_container_suffix }}'
repo_gpgcheck: '{{ libnvidia_container_repo_gpgcheck }}'
gpgcheck: '{{ libnvidia_container_gpgcheck }}'
enabled: 1
gpgkey: https://nvidia.github.io/libnvidia-container/gpgkey
sslverify: 1
sslcacert: /etc/pki/tls/certs/ca-bundle.crt
state: present

- name: nvidia-container-runtime repo
yum_repository:
name: nvidia-container-runtime
description: nvidia-container-runtime
baseurl: '{{ nvidia_container_hook_repo_base_url }}{{ nvidia_container_runtime_suffix }}'
repo_gpgcheck: '{{ nvidia_container_runtime_repo_gpgcheck }}'
gpgcheck: '{{ nvidia_container_runtime_gpgcheck }}'
enabled: 1
gpgkey: https://nvidia.github.io/nvidia-container-runtime/gpgkey
sslverify: 1
sslcacert: /etc/pki/tls/certs/ca-bundle.crt
state: present

- name: install nvidia-container-runtime-hook
yum: name=nvidia-container-runtime-hook state=latest
Expand Down
2 changes: 2 additions & 0 deletions playbooks/roles/nvidia-driver-install/defaults/main.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
cuda_driver_repo_base_url: 'http://developer.download.nvidia.com/'
cuda_driver_repo_suffix: 'compute/cuda/repos/rhel7/x86_64'
11 changes: 9 additions & 2 deletions playbooks/roles/nvidia-driver-install/tasks/main.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,15 @@
- name: install EPEL
yum: name=https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm state=present

- name: install NVIDIA repo
yum: name=https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-9.1.85-1.x86_64.rpm state=present
- name: cuda NVIDIA repository
yum_repository:
name: cuda
description: cuda
baseurl: '{{ cuda_driver_repo_base_url }}{{ cuda_driver_repo_suffix }}'
enabled: 1
gpgcheck: 1
gpgkey: http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/7fa2af80.pub
state: present

- name: install NVIDIA drivers
yum: name={{item}} state=present
Expand Down