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

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