torch_geometric/pyg 사용 시 propagate 혹은 forward 오류 해결법

Graph neural network 구현을 용이하도록 개발된 torch_geometric 혹은 요즘 새로운 이름 pyg를 사용할 때, 모든 오류를 고쳐서 가까스로 import 했더니 propagate나 forward 에서 터져버리는 오류가 날 때가 있다. 

 

온갖 구글링 결과, 같은 상황으로 고통받던 한 사람이 단순히 open3d와 torch의 import 순서를 바꿨더니 해결했다는 말이 있었다.

 

다시 말하면, 왜 그런지 모르겠는데 open3d가 torch 이전에 import 되어있을 때 났다

import open3d
import torch # this order causes torch_geometric error

위와 같은 순서로 하면 오류가 나고

 

import torch
import open3d # this order does not cause torch_geometric error

위와 같은 순서로 하면 오류가 '안' 난다.

 

예전에도 한 번 torch import 순서 때문에 시간을 날린 적이 있는데, 맘 편하려면 모든 코드에서 torch를 사용할 경우 제일 상단에서 import 하는게 좋을 것 같다.

 

정말 뭐 같은 오류다.