--- # tasks file for nvidia-proprietary - name: Add contrib & non-free repository replace: dest: /etc/apt/sources.list regexp: '^(deb(?!.* contrib).*)' replace: '\1 contrib non-free' - name: Install Linux headers apt: name: linux-headers-{{ ansible_kernel }} state: present - name: Download Nvidia driver local repo get_url: url: https://us.download.nvidia.com/tesla/550.90.07/nvidia-driver-local-repo-debian12-550.90.07_1.0-1_amd64.deb dest: /tmp/nvidia-driver-local-repo-debian12-550.90.07_1.0-1_amd64.deb - name: Install Nvidia driver local repo apt: deb: /tmp/nvidia-driver-local-repo-debian12-550.90.07_1.0-1_amd64.deb state: present - name: Add Nvidia driver local repo keyring copy: remote_src: true src: /var/nvidia-driver-local-repo-debian12-550.90.07/nvidia-driver-local-3FEEC8FF-keyring.gpg dest: /usr/share/keyrings/nvidia-driver-local-3FEEC8FF-keyring.gpg - name: Download CUDA repo get_url: url: https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda-repo-debian12-12-4-local_12.4.1-550.54.15-1_amd64.deb dest: /tmp/cuda-repo-debian12-12-4-local_12.4.1-550.54.15-1_amd64.deb - name: Install CUDA repo apt: deb: /tmp/cuda-repo-debian12-12-4-local_12.4.1-550.54.15-1_amd64.deb state: present - name: Add CUDA repo keyring copy: remote_src: true src: /var/cuda-repo-debian12-12-4-local/cuda-C5AA6424-keyring.gpg dest: /usr/share/keyrings/cuda-C5AA6424-keyring.gpg - name: Update package list apt: update_cache: yes - name: Install Nvidia driver and CUDA toolkit apt: name: - firmware-misc-nonfree - cuda-toolkit-12-4 - nvidia-driver=550.90.07-1 state: present register: install_driver - name: Reboot after driver install reboot: when: install_driver.changed and not nvidia_driver_skip_reboot