Update installation to use torch, PyG, lightning >= 2

Created by: asogaard

The changes made to PyG >= 2.0 should make the installation process much simpler.

Depends on #477 (closed)