ubuntu 22

tasks/nvidia.yml

@@ -45,14 +45,19 @@
   find:
     paths:
       - /usr/lib/x86_64-linux-gnu
-    patterns: 'libcudnn.so.7.6.5'
+    patterns: 'libcudnn.so.8.4.0'
   register: cudnn_result
+- name: purge cudnn7
+  become: true
+  apt:
+    name: libcudnn7,libcudnn7-dev,libcudnn7-doc
+    state: absent
 - name: install cudnn
   become: true
   apt:
     deb: 'https://cdn02.moecube.com:444/init/cudnn/{{item}}'
   with_items:
-    - libcudnn8_7.6.5.32-1+cuda10.1_amd64.deb
-    - libcudnn8-dev_7.6.5.32-1+cuda10.1_amd64.deb
-    - libcudnn8-samples_7.6.5.32-1+cuda10.1_amd64.deb
-  when: not cudnn_result.files[0] is defined and ansible_distribution == "Ubuntu" and ansible_distribution_major_version|int >= 18
+    - libcudnn8_8.4.0.27-1+cuda11.6_amd64.deb
+    - libcudnn8-dev_8.4.0.27-1+cuda11.6_amd64.deb
+    - libcudnn8-samples_8.4.0.27-1+cuda11.6_amd64.deb
+  when: not cudnn_result.files[0] is defined and ansible_distribution == "Ubuntu" and ansible_distribution_major_version|int >= 20

tasks/utility.yml

@@ -76,9 +76,9 @@
   set_fact:
     target_distribution: '{{ ansible_distribution_release }}'
   when: not upgrade or ansible_distribution_release == 'sid'
-- name: move to focal
+- name: move to jammy
   set_fact:
-    target_distribution: 'focal'
+    target_distribution: 'jammy'
   when: ansible_distribution == 'Ubuntu' and upgrade
 - name: move to bullseye
   set_fact: