torch-geometric error

OSError: []/python3.7/site-packages/torch_sparse/_version.so: undefined symbol: _ZN3c1017RegisterOperatorsC1Ev

위와 같은 에러가 나타나서 torch_sparse를 설치하고 그랬는데 안 되어서 검색해본 결과, version이 안 맞는걸 확인하였다. 이럴 때는 처음부터 다시 conda의 환경을 생성하면 된다. 

pytorch-geometric.readthedocs.io/en/latest/notes/installation.html에 접속하여 코드를 작성하면 되는데,

Installation — pytorch_geometric 1.7.0 documentation

1) 토치 버전과 쿠다 버전을 확인한다.

import torch
print(torch.__version__)
print(torch.version.cuda)

2) pytorch-geometric.com/whl/ 에 접속하여 지원하는 torch버전과 cuda버전을 확인하고  

3) 아래 코드를 순서대로 입력한다.

TORCH = [해당 torch version입력]
CUDA = [해당 cuda version 입력]
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

그러면 잘 돌아간다