classroom/roles/linroom/tasks/pytorch.yml

9 lines
134 B
YAML
Raw Normal View History

2023-09-12 12:14:11 +00:00
- name: Install pytorch
apt:
name:
- python3-torch
- python3-skorch
- python3-torch-ignite
state: latest