diff --git a/roles/nvidia/tasks/main.yml b/roles/nvidia/tasks/main.yml index 9d9be68..007e5dd 100644 --- a/roles/nvidia/tasks/main.yml +++ b/roles/nvidia/tasks/main.yml @@ -47,43 +47,17 @@ state: present -- name: Recolher lista de pacotes instalados - become: yes - ansible.builtin.package_facts: - manager: apt - -- name: Configurar runtime Docker - become: yes - ansible.builtin.shell: | - nvidia-ctk runtime configure --runtime=docker - when: "'docker.io' in ansible_facts.packages" - -- name: Reiniciar Docker - become: yes - ansible.builtin.systemd: - name: docker - state: restarted - when: "'docker.io' in ansible_facts.packages" - - name: Configurar runtime containerd become: yes ansible.builtin.shell: | nvidia-ctk runtime configure --runtime=containerd - when: "'containerd' in ansible_facts.packages" - name: Reiniciar containerd become: yes ansible.builtin.systemd: name: containerd state: restarted - when: "'containerd' in ansible_facts.packages" -- name: Validar GPU dentro de container - become: yes - ansible.builtin.shell: | - docker run --rm --runtime=nvidia --gpus all nvidia/cuda:12.2-base nvidia-smi - register: nvidia_smi_output - ignore_errors: true - name: Mostrar resultado nvidia-smi become: yes