disable sleep

tasks/main.yml

 ---
 # tasks file for init
-- name: main tasks
-  include_tasks: '{{item}}.yml'
-  with_items:
-    - utility
-    - sshd_config
+- name: utility tasks
+  include_tasks: 'utility.yml'
+- name: sshd_config
+  include_tasks: 'sshd_config.yml'
 - name: detect swap
   shell: 'swapon'
   register: swap_result

tasks/nvidia.yml

@@ -38,3 +38,18 @@
     - libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64
     - libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64
   when: not cudnn_result.files[0] is defined and ansible_distribution == "Ubuntu" and ansible_distribution_major_version|int >= 18
+- name: nvidia 重启
+  become: true
+  shell: reboot
+  async: 0
+  poll: 0
+  register: reboot_result
+  args:
+    removes: /var/run/reboot-required
+  when: reboot
+- name: nvidia 重启等待
+  local_action: wait_for host={{ ansible_ssh_host }} state=started
+  when: reboot and reboot_result.changed
+- name: 重新读取主机信息
+  setup:
+  when: reboot and reboot_result.changed

tasks/utility.yml

@@ -54,6 +54,18 @@
   become: true
   timezone:
     name: Asia/Shanghai
+- name: disable auto sleep
+  become: true
+  file:
+    path: '/etc/systemd/system/{{item}}.target'
+    state: link
+    src: /dev/null
+    force: true
+  with_items:
+    - sleep
+    - suspend
+    - hibernate
+    - hybrid-sleep
 - name: set facts for China mirror
   set_fact:
     mirror_debian: http://mirrors.tuna.tsinghua.edu.cn