【windows】安装pytorch1.7.1-cpu版本的geometric过程记录

首先在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)

这里我们创建了一个自定义的数据,输出结果:
在这里插入图片描述
说明没有问题了

猜你喜欢

转载自blog.csdn.net/vincent_duan/article/details/120463046