Add requirements for Podatkovno Rudarjenje

roles/torch/tasks/main.yml

@@ -0,0 +1,4 @@
+- import_tasks: main_win.yml
+  when: ansible_os_family == 'Windows'
+- import_tasks: main_lin.yml
+  when: ansible_os_family == 'Debian'

roles/torch/tasks/main_lin.yml

@@ -0,0 +1,12 @@
+- name: Install torch
+  apt:
+    name:
+      - python3-torch
+      - python3-skorch
+      - python3-torch-cluster
+      - python3-torch-geometric
+      - python3-ignite
+      - python3-sparse
+      - python3-torchaudio
+      - python3-torchvision
+    state: latest

roles/torch/tasks/main_win.yml

@@ -0,0 +1,2 @@
+- name: Install torch
+  win_command: pip install torch torchvision