首先在https://pytorch-geometric.com/whl/,查看对应版本的whl文件,这里进入torch-1.7.1+cpu
,然后下载,如下图所示:
选择对应的文件:
torch_scatter-2.0.7-cp36-cp36m-win_amd64.whl
torch_sparse-0.6.9-cp36-cp36m-win_amd64.whl
torch_spline_conv-1.2.1-cp36-cp36m-win_amd64.whl
torch_cluster-1.5.9-cp36-cp36m-win_amd64.whl
将whl文件下载到电脑上,使用pip进行安装,过程如下:
最后使用pip install torch-geometric
命令安装即可,下面展示部分截图:
简单测试一下torch_geometric
:
import torch
from torch_geometric.data import Data
edge_index = torch.tensor([
[0,1,1,2],
[1,0,2,1]
], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)
data = Data(x=x, edge_index=edge_index)
这里我们创建了一个自定义的数据,输出结果:
说明没有问题了