论文需要做新模型、新案例,将原来做的东西又捡起来,倒腾倒腾。

PyTorch Geometric Temporal 是 PyTorch Geometric 的一个时序图神经网络扩展库。它建立在开源深度学习和图处理库之上。PyTorch Geometric Temporal 包含最先进的深度学习和参数学习方法来处理时空信号。它是第一个用于几何结构上的时序深度学习的开源库,并提供针对动态和静态图的常数时间差分图神经网络。我们通过使用离散时图快照来实现这一点。实施的方法涵盖了广泛的数据挖掘 (WWW, KDD)、人工智能和机器学习 (AAAI, ICONIP, ICLR) 会议、研讨会和知名期刊的论文。———官方介绍

报错

当采用 pip 安装时,无任何错误提示,但当导入 torch_geometric_temporal 时报错:

1
ModuleNotFoundError: No module named 'torch_geometric.utils.to_dense_adj'

解决方案

研究发现torch_geometric_temporal项目已经太久没更新,版本定格于 0.54.0,更新于 2022 年 9 月 5 日。

而其所依赖的torch_geometric项目却一直在更新,最新版为 2.5.0,发布于 2024 年 2 月 16 日。

故将torch_geometric版本回退到 2.2.0 版,问题解决。

附-安装方案-CUDA 11.8

CUDA

安装 CUDA 和 CUDNN 库,从 NVIDIA 官网安装,选择合适的版本(目前用的 11.8)

PyTorch

安装 PyTorch,按官方推荐方式安装

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

PyTorch Geometric

安装 PyTorch Geometric,官方推荐方式,注意,要安装旧版,推荐安装 2.2.0 版,实测与torch_geometric_temporal==0.54.0兼容。

1
2
3
4
pip install torch_geometric==2.2.0

# Optional dependencies:
pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cu118.html

PyTorch Geometric Temporal

安装 PyTorch Geometric Temporal,直接使用 pip 即可:

1
pip install torch_geometric_temporal