main.yml 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. ---
  2. - name: install Nvidia driver
  3. yum:
  4. name:
  5. - kmod-nvidia
  6. #- nvidia-x11-drv
  7. state: present
  8. tags: install
  9. #- name: add Nvidia container runtime support
  10. #get_url:
  11. #url: https://nvidia.github.io/nvidia-docker/centos7/nvidia-docker.repo
  12. #dest: /etc/yum.repos.d/nvidia-docker.repo
  13. #tags: install
  14. - name: add Nvidia container runtime support
  15. get_url:
  16. url: https://nvidia.github.io/nvidia-container-runtime/centos7/nvidia-container-runtime.repo
  17. dest: /etc/yum.repos.d/nvidia-container-runtime.repo
  18. tags: install, testing
  19. # disable gpg key (because Nvidia doesn't know how to make that work yet for some reason)
  20. - replace:
  21. path: /etc/yum.repos.d/nvidia-container-runtime.repo
  22. regexp: 'repo_gpgcheck=1'
  23. replace: 'repo_gpgcheck=0'
  24. backup: yes
  25. tags: testing
  26. - name: install Nvidia-container-runtime-hook
  27. yum:
  28. name:
  29. #- nvidia-detect
  30. #- kmod-nvidia-410.73-1.el7_5.elrepo
  31. - nvidia-container-runtime-hook
  32. state: present
  33. tags: install
  34. # This needs to be done on GPU nodes
  35. #- name: Enable DevicePlugins for all GPU nodes (nvidia-container-runtime-hook)
  36. #copy: src=nvidia dest=/usr/libexec/oci/hooks.d/ owner=root group=root mode=755
  37. #tags: install
  38. #- name: Add KUBE_EXTRA_ARGS to enable Plugins (GPU support) --III alreday done in common
  39. #lineinfile:
  40. #path: /etc/systemd/system/kubelet.service.d/10-kubeadm.conf
  41. #line: 'Environment="KUBELET_EXTRA_ARGS=--feature-gates=DevicePlugins=true"'
  42. #insertbefore: 'KUBELET_KUBECONFIG_ARGS='
  43. #tags: install
  44. - name: Restart and Enable docker service
  45. service:
  46. name: docker
  47. state: restarted
  48. enabled: yes
  49. tags: install
  50. - name: Restart and Enable Kubernetes - kubelet
  51. service:
  52. name: kubelet
  53. state: restarted
  54. enabled: yes
  55. tags: install