ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

windows安装CPU版本 torch_geometric教程

2021-10-01 21:58:32  阅读:472  来源: 互联网

标签:genmetric windows torch 如下 sparse 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])

标签:genmetric,windows,torch,如下,sparse,geometric,安装
来源: https://blog.csdn.net/liaoningxinmin/article/details/120581075

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有