一、图神经网络环境配置
·
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
这些版本经过我的尝试是兼容的
更多推荐
所有评论(0)