PyG初探(一)

今天主要安装了一下torch-geometric,本来之前以为直接pip install就完事了,结果使用时发现了很多报错,主要报错有
1:undefined symbol: _ZN3c105ErrorC1ENS_14SourceLocationERKSs
2:ModuleNotFoundError: No module named ‘torch_scatter’
3:AttributeError: ‘NoneType‘ object has no attribute ‘origin‘

其实这些报错都是一个问题,包版本不对。主要涉及的包有torch,cuda,torch-sparse,torch-scatter,torch-geometric。没有torch-scatter会报错误2,torch-scatter版本不对会报错误1,torch-sparse版本不对会报错误3。

解决方法:首先检查torch和cuda的版本对应,这个应该不会有什么问题。然后如果之前用pip安装了geometric,先卸载已安装的3个包,按照官网指令重新安装https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html,就不会出现问题了。

你可能感兴趣的:(深度学习,深度学习)