[pytorch] ModuleNotFoundError: No module named 'torch_geometric.data.dataloader'

베이스 코드를 그대로 돌렸을 때, 아예 모듈을 찾지 못하는 경우가 종종 있습니다.

대부분 버전이 맞지 않을 경우의 문제이지만, 매번 버전을 확인하는 것도 귀찮은 일입니다.

 

이 경우 pytorch-geometric 의 data.dataloader 를 가져오지 못하는 문제가 발생합니다.

경험상 이런 경우 그냥 바로 공식 문서로 가서 확인하는게 제일 빠르게 해결되는 것 같네요. 

 

공식문서에 따르면, torch_geometric.data.dataloader 가 아닌 torch_geomteric.loader.dataloader에 속한 것을 확인 할 수 있고, 이 부분을 바꿔주면서 문제가 해결되었습니다. 

 

https://pytorch-geometric.readthedocs.io/en/latest/_modules/torch_geometric/loader/dataloader.html#DataLoader

torch_geometric.loader.dataloader — pytorch_geometric documentation