一、图神经网络环境配置
·
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
这些版本经过我的尝试是兼容的
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)