snap-stanford / stellar

MIT License
61 stars 17 forks source link

not all requirements are listed in requirements.txt #2

Closed igordot closed 1 year ago

igordot commented 2 years ago

I am trying to use STELLAR. I noticed not all requirements are listed in requirements.txt. I had to add torch_geometric.

MonoHue commented 1 year ago

I think torch_geometric should be installed through conda install pyg -c pyg. Using pip to install it may have difficulty in building wheels.

johnF-moore commented 1 year ago

I ran into the same issue. The problem I had was that torch_geometric only supports CUDA, so if you don't use NVIDIA GPUs, then you have to install the CPU-only versions of pytorch and pyg. Here's the relevant issue from the pytorch_geometric github: https://github.com/pyg-team/pytorch_geometric/issues/837.

This is what I used to get the demo to work.

conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pyg -c pyg