nvidia patch

tasks/nvidia.yml

@@ -22,7 +22,25 @@
   become: true
   apt:
     update_cache: true
-    name: 'nvidia-driver-{{ nvidia_driver_version }},nvidia-cuda-dev,nvidia-cuda-doc,nvidia-cuda-gdb,nvidia-cuda-toolkit,nvidia-cuda-toolkit-gcc,nvidia-container-runtime,libnvidia-encode-{{ nvidia_driver_version }},libnvidia-decode-{{ nvidia_driver_version }},libnvidia-fbc1-{{ nvidia_driver_version }}'
+    name: 'git,nvidia-driver-{{ nvidia_driver_version }},nvidia-cuda-dev,nvidia-cuda-doc,nvidia-cuda-gdb,nvidia-cuda-toolkit,nvidia-cuda-toolkit-gcc,nvidia-container-runtime,libnvidia-encode-{{ nvidia_driver_version }},libnvidia-decode-{{ nvidia_driver_version }},libnvidia-fbc1-{{ nvidia_driver_version }}'
+- name: check detailed driver version
+  changed_when: false
+  shell: nvidia-smi --query-gpu=driver_version --format=csv,noheader,nounits
+  register: driver_version_result
+- name: nvidia-patch repo
+  git:
+    dest: /tmp/nvidia-patch
+    repo: https://code.mycard.moe/nanahira/nvidia-patch
+- name: patch
+  include_tasks: nvidia_patch.yml
+  vars:
+    driver_detail_version: '{{driver_version_result.stdout}}'
+    patch_info: '{{item}}'
+  with_items:
+    - script: patch-fbc.sh
+      check_file: libnvidia-fbc-backup/libnvidia-fbc.so
+    - script: patch.sh
+      check_file: libnvidia-encode-backup/libnvidia-encode.so
 - name: check install of cudnn
   find:
     paths:

tasks/nvidia_patch.yml

+- name: check patch
+  changed_when: false
+  failed_when: patch_check_result.rc > 1
+  register: patch_check_result
+  shell: /tmp/nvidia-patch/{{patch_info.script}} -c {{driver_detail_version}}
+- name: patch driver
+  become: true
+  when: patch_check_result.rc == 0
+  shell: /tmp/nvidia-patch/{{patch_info.script}} -d {{driver_detail_version}}
+  args:
+    creates: /opt/nvidia/{{patch_info.check_file}}.{{driver_detail_version}}