Add tasks

local.yml

----
-- hosts: localhost
-  pre_tasks:
-    - name: Upgrade packages
-      become: true
-      apt:
-        upgrade: true
-        update_cache: true
-        force_apt_get: true
-
-  roles:
-    - jupyter-lab
-    - ml-python-packages
-    - misc-applications
+---
+- hosts: localhost
+  pre_tasks:
+    # - name: Upgrade packages
+    #   become: true
+    #   apt:
+    #     upgrade: true
+    #     update_cache: true
+    #     force_apt_get: true
+
+    # - name: Install pip3
+    #   become: true
+    #   apt:
+    #     update_cache: true
+    #     name:
+    #       - python3-pip
+    #       - libffi-dev # This is needed for jupyter on aarch64 to compile its C backend
+    #     state: present
+
+  roles:
+    - jupyter-lab
+    - ml-python-packages
+    - misc-applications

roles/jupyter-lab/files/jupyter.service

+[Unit]
+Description=Jupyter Lab
+
+[Service]
+Type=simple
+PIDFile=/run/jupyter.pid
+# Password: [empty string]
+ExecStart=/usr/bin/python3 -m jupyterlab --ip="0.0.0.0" --notebook-dir=/home/pi/notebooks --no-browser --NotebookApp.password='sha1:9a2d316959ac:843b251c27024afb46174ce40ce0ebebcf29217b'
+User=pi
+Group=pi
+Restart=always
+RestartSec=10
+
+[Install]
+WantedBy=multi-user.target

roles/jupyter-lab/tasks/main.yml

----
-# roles/jupyter-lab
+---
+# roles/jupyter-lab
+
+- name: Notebooks dir
+  file:
+    path: /home/pi/notebooks
+    state: directory
+    owner: pi
+    group: pi
+
+# This should be optimized so that only one check / install is needed
+# Still we should make sure that pip3 is actually installed before trying to use it
+- name: Install dependecies
+  become: true
+  apt:
+    name:
+      - python3-pip
+    state: present
+    update_cache: true
+
+- name: Install jupyter-lab
+  become: true
+  pip:
+    executable: pip3
+    name: jupyterlab
+    state: present
+
+- name: Install systemd service
+  become: true
+  copy:
+    src: files/jupyter.service
+    dest: /etc/systemd/system/jupyter.service
+    mode: '0644'
+  
+- name: Start Jupyter-Lab service
+  become: true
+  systemd:
+    name: jupyter
+    state: started
+    enabled: true
+    daemon_reload: true

roles/misc-applications/tasks/main.yml

----
-# roles/misc-applications
+---
+# roles/misc-applications

roles/ml-python-packages/tasks/main.yml

----
-# roles/ml-python-packages
+---
+# roles/ml-python-packages
+
+# This should be optimized so that only one check / install for pip3 is needed
+# Still we should make sure that pip3 is actually installed before trying to use it
+- name: Install dependecies
+  become: true
+  apt:
+    name:
+      - python3-pip
+      - libatlas-base-dev
+    state: present
+    update_cache: true
+
+- name: Remove conflicting apt packages
+  become: true
+  apt:
+    name:
+      - python3-pil
+      - python3-numpy
+    state: absent
+
+- name: Install ml python packages
+  become: true
+  pip:
+    executable: pip3
+    name:
+      - numpy
+      - sklearn
+      - matplotlib
+      - pandas
+      - Pillow
+      # - torch
+      # - torchvision
+      # - fastai
+    state: present