benedekrozemberczki/SimGNN

Import error for scatter_cuda

jonathan-goh opened this issue · 1 comments

jonathan-goh commented

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

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)