diff --git a/roles/nvidia/tasks/main.yml b/roles/nvidia/tasks/main.yml
index 65877d8..9d9be68 100644
--- a/roles/nvidia/tasks/main.yml
+++ b/roles/nvidia/tasks/main.yml
@@ -1,11 +1,15 @@
-- name: Instalar as nvidia drivers
+- name: Instalar driver NVIDIA
   become: yes
-  become_user: fenix 
-  shell: |
-    sudo apt install nvidia-driver-550
-    sudo apt install nvidia-utils-550
-  environment:
-    KUBECONFIG:  /home/fenix/.kube/config
+  ansible.builtin.apt:
+    name: nvidia-driver-550
+    state: present
+
+- name: Instalar utilitários NVIDIA
+  become: yes
+  ansible.builtin.apt:
+    name: nvidia-utils-550
+    state: present
+
 
 - name: Reboot host e esperar voltar
   become: yes