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