1.安装pytorch包

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

2.安装torch-geometric

pip install torch-geometric==2.0.4

3.下载四个配置文件cluster scatter sparse spline
访问:https://data.pyg.org/whl/
下载对应torch版本1.12.1和cu113对应的cluster scatter sparse spline二进制文件

安装命令为 pip install torch_cluster 文件路径


下面是一个torch_cluster对应的安装示例:

pip install torch_cluster /home/my/gnn/learn/torch_scatter-2.1.0+pt112cu113-cp310-cp310-linux_x86_64.whl

4.安装torch_geometric_temporal

其实也可以根据安装torch_geometric_temporal报错再安装3中的文件,如图所示:

pip install torch_geometric_temporal==0.54.0

另外,给出参考:
torch-geometric的github链接:

https://github.com/pyg-team/pytorch_geometric?tab=readme-ov-file
torch_geometric_temporal的github链接:

https://github.com/benedekrozemberczki/pytorch_geometric_temporal/tree/master/dataset
这些版本经过我的尝试是兼容的

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐