12 lines
265 B
YAML
12 lines
265 B
YAML
- name: Install torch
|
|
apt:
|
|
name:
|
|
- python3-torch
|
|
- python3-skorch
|
|
- python3-torch-cluster
|
|
- python3-torch-geometric
|
|
- python3-ignite
|
|
- python3-sparse
|
|
- python3-torchaudio
|
|
- python3-torchvision
|
|
state: latest
|