ModuleNotFoundError: No module named 'torch_scatter'

https://towardsdatascience.com/hands-on-graph-neural-networks-with-pytorch-pytorch-geometric-359487e221a8

Hands on Graph Neural Networks with PyTorch & PyTorch Geometric

위 튜토리얼을 따라하려는데 에러가 발생했다.

 

Solution

찾아보니까 torch_geometric 외에도 다운받아 줘야 하는 게 있어서 torch version이랑 geometric version 맞춰서 다시 install 진행함

 

torch install

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

torch geometric install 

(https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html 참고)

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cu102.html