- 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