classroom/roles/pytorch/tasks/main_lin.yml

9 lines
134 B
YAML

- name: Install pytorch
apt:
name:
- python3-torch
- python3-skorch
- python3-torch-ignite
state: latest