windows安装CPU版本 torch_geometric教程

本文主要是介绍windows安装CPU版本 torch_geometric教程，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

由于最近需要搞图神经网络（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])

这篇关于windows安装CPU版本 torch_geometric教程的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！