Skip to content
Snippets Groups Projects

Adding Nvidia gpu-operator installation

Merged Jaromír Hradil requested to merge nvidia_gpu_install into master
+ 6
5
@@ -588,21 +588,22 @@
shell: |-
helm repo add nvidia https://helm.ngc.nvidia.com/nvidia
helm repo update
changed_when: true
when: "'nvidia' not in ansible_local.helm_repos | map(attribute='name') | list"
- name: Deploy/upgrade Nvidia gpu-operator instance
vars:
version: "24.9.2"
driver_enabled: false # if set to true the operator will install containerized drivers
mig_strategy: single
shell: |-
helm status --namespace gpu-operator gpu-operator
if [ $? -ne 0 ]; then
helm install --wait --create-namespace --namespace gpu-operator \
--version {{ version }} gpu-operator nvidia/gpu-operator \
--set driver.enabled={{ driver_enabled }} --set mig.strategy={{ mig_strategy }}
gpu-operator nvidia/gpu-operator --set driver.enabled={{ driver_enabled }} \
--set mig.strategy={{ mig_strategy }}
else
helm upgrade --wait --namespace gpu-operator \
--version {{ version }} gpu-operator nvidia/gpu-operator \
--set driver.enabled={{ driver_enabled }} --set mig.strategy={{ mig_strategy }}
gpu-operator nvidia/gpu-operator --set driver.enabled={{ driver_enabled }} \
--set mig.strategy={{ mig_strategy }}
fi
changed_when: true
when: true
Loading