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])
标签:genmetric,windows,torch,如下,sparse,geometric,安装 来源: https://blog.csdn.net/liaoningxinmin/article/details/120581075