Initial commit erasing history

roles/pytorch/tasks/main.yml

@@ -0,0 +1,4 @@
+- import_tasks: main_win.yml
+  when: ansible_connection == 'winrm'
+- import_tasks: main_lin.yml
+  when: ansible_connection == 'ssh'

roles/pytorch/tasks/main_lin.yml

@@ -0,0 +1,8 @@
+- name: Install pytorch
+  apt:
+    name:
+      - python3-torch
+      - python3-skorch
+      - python3-torch-ignite
+    state: latest
+

roles/pytorch/tasks/main_win.yml

@@ -0,0 +1,2 @@
+- name: Install pytorch
+  win_command: pip install torch skorch pytorch-ignite