Torchgraphs
A PyTorch library for Graph Convolutional Networks.
Requirements and installation
Torchgraphs is developed in Python 3.7 and depende on PyTorch 1.1. It is suggested but not required to install PyTorch beforehand, to correctly match the hardware capabilities, see the official installation instruction.
All requirements listed in requirements.txt will be installed automatically when running:
pip install https://github.com/baldassarreFe/torchgraphs
To develop the library itself, a conda environment with additional dependencies is provided, as well as a test suite for pytest:
git clone https://github.com/baldassarreFe/torchgraphs
cd torchgraphs
ENV_NAME=torchgraphs_env
conda env create -n "${ENV_NAME}" -f conda.yaml
conda activate "${ENV_NAME}"
pip install --editable .
pytest