8 lines
134 B
YAML
8 lines
134 B
YAML
- name: Install pytorch
|
|
apt:
|
|
name:
|
|
- python3-torch
|
|
- python3-skorch
|
|
- python3-torch-ignite
|
|
state: latest
|
|
|