diff --git a/roles/nvidia/tasks/main.yml b/roles/nvidia/tasks/main.yml index c21ce2b..4ee93cf 100644 --- a/roles/nvidia/tasks/main.yml +++ b/roles/nvidia/tasks/main.yml @@ -33,13 +33,11 @@ - name: Atualizar cache de pacotes become: yes - become_user: fenix ansible.builtin.apt: update_cache: yes - name: Instalar NVIDIA Container Toolkit become: yes - become_user: fenix ansible.builtin.apt: name: nvidia-container-toolkit state: present @@ -47,14 +45,12 @@ - name: Configurar runtime Docker become: yes - become_user: fenix ansible.builtin.shell: | nvidia-ctk runtime configure --runtime=docker when: "'docker.io' in ansible_facts.packages" - name: Reiniciar Docker become: yes - become_user: fenix ansible.builtin.systemd: name: docker state: restarted @@ -62,14 +58,12 @@ - name: Configurar runtime containerd become: yes - become_user: fenix ansible.builtin.shell: | nvidia-ctk runtime configure --runtime=containerd when: "'containerd' in ansible_facts.packages" - name: Reiniciar containerd become: yes - become_user: fenix ansible.builtin.systemd: name: containerd state: restarted @@ -77,7 +71,6 @@ - name: Validar GPU dentro de container become: yes - become_user: fenix ansible.builtin.shell: | docker run --rm --runtime=nvidia --gpus all nvidia/cuda:12.2-base nvidia-smi register: nvidia_smi_output @@ -85,6 +78,5 @@ - name: Mostrar resultado nvidia-smi become: yes - become_user: fenix ansible.builtin.debug: var: nvidia_smi_output.stdout \ No newline at end of file