Skip to content

Commit 3e9ab24

Browse files
committed
Add option to install nvidia-fabricmanager
1 parent 81a2581 commit 3e9ab24

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

ansible/roles/cuda/defaults/main.yml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,17 +2,23 @@
22
# yamllint disable-line rule:line-length
33
cuda_repo_url: "https://developer.download.nvidia.com/compute/cuda/repos/rhel{{ ansible_distribution_major_version }}/{{ ansible_architecture }}/cuda-rhel{{ ansible_distribution_major_version }}.repo"
44
cuda_nvidia_driver_stream: '580-open'
5-
cuda_nvidia_driver_pkg: "nvidia-open-3:580.82.07-1.el{{ ansible_distribution_major_version }}"
5+
cuda_nvidia_driver_version: '580.82.07-1'
6+
cuda_nvidia_driver_pkg: "nvidia-open-3:{{ cuda_nvidia_driver_version }}.el{{ ansible_distribution_major_version }}"
67
cuda_package_version: '13.0.1-1'
78
cuda_version_short: "{{ (cuda_package_version | split('.'))[0:2] | join('.') }}" # major.minor
8-
cuda_packages:
9+
cuda_packages_default:
910
- "cuda-toolkit-{{ cuda_package_version }}"
1011
- nvidia-gds
1112
- cmake
13+
cuda_packages_fabricmanager:
14+
- "nvidia-fabricmanager-{{ cuda_nvidia_driver_version }}"
15+
cuda_packages: "{{ cuda_packages_default + ( cuda_packages_fabricmanager if cuda_install_nvidiafabricmanger | bool else [] ) }}"
1216
cuda_samples_release_url: "https://github.com/NVIDIA/cuda-samples/archive/refs/tags/v{{ cuda_version_short }}.tar.gz"
1317
cuda_samples_path: "/var/lib/{{ ansible_user }}/cuda_samples"
1418
cuda_samples_programs:
1519
- deviceQuery
1620
- bandwidthTest
1721
# cuda_devices: # discovered from deviceQuery run
1822
cuda_persistenced_state: started
23+
# option whether or not to install nvidia-fabricmanager default false
24+
cuda_install_nvidiafabricmanger: false

0 commit comments

Comments
 (0)