classroom/roles/pytorch/tasks/main_lin.yml

9 lines
134 B
YAML
Raw Normal View History

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