想使用Pytorch Geometric实现GCN、GraphSAGE和GAT,首先模块直接报错。
安装步骤:
1 打开当前环境所在的命令行,输入python
查看python版本;输入import torch
和torch__version__ #双下斜杠
查看torch版本;输入torch.version.cuda
查看cuda版本。
2 下载安装所需组件链接:
https://pytorch-geometric.com/whl/
cp后面的数字代表是python版本,将
torch_cluster
、torch_scatte
r、torch_sparse
、torch_spline_conv
都下下来,放到刚刚打开python的命令行文件位置。此处我的Python版本是3.8.5,cp选择38;打开python的命令行文件位置为 C:\Users\XXX\Python
,因此放到 C:\Users\XXX
。
3 pip install 四个文件名.whl
4 安装pyg
pip install torch_geometric
3步安装torch_geometric
于 2023-03-10 16:46:11 首次发布