最近搭建gnn需要用到torch-geometric和torch,这俩常常会出现版本不兼容问题(顺带一提,我没有使用cuda,用的是cpu,所以要使用cuda的话还请注意cuda与torch的兼容性问题)
报错原因
1.5.0版本的torch不存在UninitializedParameter的属性,而我查阅1.9.1版本的torch是存在这个属性的,但是torch-geometric 2.0.3(目前的最新版)不支持1.9.1版本的torch,所以尝试选择降低torch-geometric版本为1.7,降低之后运行成功。
报错时的环境版本:
torch 1.5.0
torch-geometric 2.0.3
torch-geometric依赖包(对应torch1.5.0版本)
解决方案:
使用下面这套环境
torch 1.9.1下载地址
torch-geometric 1.7.2 下载地址
torch-geometric依赖包(对应torch1.9.1版本)下载地址
其中依赖包有四个,每种下载一个版本就行,详细安装可参考我的上一篇文章