search

benedekrozemberczki / SimGNN

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).
GNU General Public License v3.0
759 stars 147 forks source link

Import error for scatter_cuda #26

Closed jonathan-goh closed 4 years ago

jonathan-goh commented 4 years ago

Hi there,

I am having a “no module named touchy_scatter .scatter_cuda” import error. I am sure I have the necessary toolkits and driver installed.

Does the code require a GPU environment?

jonathan-goh commented 4 years ago

This was solved by installing the following:

pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

where ${CUDA} is cpu, cu92 (CUDA 9.,2), cu100 (CUDA 10) or cu101 (CUDA 10.1)