windows安装CPU版本 torch_geometric教程

  • Post author:
  • Post category:其他


由于最近需要搞图神经网络(GNN),因此需要安装

torch_genmetric:Graph Neural Network Library for PyTorch

,torch_genmetric 封装了图神经网络常用的算法,例如 GCN(图卷积网络)、GAT(图注意力网络)等等,因此安装其是十分必要的。


本人环境配置:系统:windows10,python版本:3.6.1,pytorch版本:1.5.0+cpu

在尝试

官网方法

安装失败后,通过

下载安装文件到本地后成功安装torch_genmetric

,现将方法记录如下:

1、torch_genmetric 安装

首先通过

官方连接

下载其安装包,本人安装为止torch_genmetric版本为

2.0.1

,如下图所示

下载后将其放置于安装python的路径下,本人具体路径为 D:\pythonsetup\Scripts(

不同python安装路径视情况而定

) 将下载的文件放置后在D:\pythonsetup\Scripts,如下图所示:

然后在 D:\pythonsetup\Scripts 路径下打开cmd 并输入

pip install torch_geometric-2.0.1.tar.gz

出现如下文字则完成安装

Successfully built torch-geometric
Installing collected packages: torch-geometric
Successfully installed torch-geometric-2.0.1

安装后通过官方代码尝试运行,代码如下:

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)
print(data)

但发现出现如下问题:

表明 torch_genmetric 的

torch_sparse

依赖库没有安装,下节中介绍torch_sparse安装方法。

2、torch_sparse 安装

torch_sparse的安装在此依然通过本地文件安装方法,在

官网

下载文件,进入官网后

首先选择torch版本

,如下图所示:

如上图为官网提供的所有torch版本,本人选择的为

torch-1.5.0+cpu

,点击进入选择torch_sparse,如下图所示:

如上图所示为官方提供的所有 torch_sparse ,cp36表示python版本为3.6,根据自己系统选择相应的文件下载并存放至 D:\pythonsetup\Scripts 文件夹,并通过第一节中介绍的方法通过pip安装。安装后继续尝试运行第一节中官方代码,出现:

表明 torch_scatter 依赖库没有安装,于是 在

官网

下载文件 ,并通过

如上方法

安装即可,torch_scatter 安装后继续尝试运行第一节中官方代码,出现:

由最后一行中的提示可以看出需要去除processed文件夹,可能是由于版本问题,本文文件存放路径如下:

本人将processed文件夹去除后继续尝试运行第一节中官方代码,终于终于成功运行,控制台输出:

Data(x=[3, 1], edge_index=[2, 4])



版权声明:本文为liaoningxinmin原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。