windows安装CPU版本 torch_geometric教程

2021/10/2 7:10:49

本文主要是介绍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教程的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程