其他分享
首页 > 其他分享> > 解决调用torch_geometric报错No module named ‘torch_sparse‘等问题,以及torch_sparse torch_scatter等的安装问题

解决调用torch_geometric报错No module named ‘torch_sparse‘等问题,以及torch_sparse torch_scatter等的安装问题

作者:互联网

出现的问题:torch_geometric报错


会出现为torch_sparse torch_scatter等的问题

最近又开始搞图神经网络方面的东西,要用到torch_geometric。但是写好代码后发现会出现报错:

Traceback (most recent call last):
  File "d:/env/CProjects/GCN/GAT.py", line 10, in <module>
  File "D:\anaconda3\lib\site-packages\torch_geometric\__init__.py", line 4, in <module>
    import torch_geometric.data
  File "D:\anaconda3\lib\site-packages\torch_geometric\data\__init__.py", line 1, in <module>
    from .data import Data
  File "D:\anaconda3\lib\site-packages\torch_geometric\data\data.py", line 3, in <module>
    from torch_geometric.typing import OptTensor, NodeType, EdgeType
  File "D:\anaconda3\lib\site-packages\torch_geometric\typing.py", line 4, in <module>
    from torch_sparse import SparseTensor
ModuleNotFoundError: No module named 'torch_sparse'

不难看出是torch_geometric中调用torch_sparse时出现了问题,而且好像是没有安装,很自然的安装这些依赖包。

如何安装依赖包

可以说这些包是真的难装,我一直会出现ERROR: Failed building wheel for torch-sparse这个问题,按我的理解来讲,这个问题有时候极有可能是包之间的版本问题引起的。然后在网上查了一下,果然是这样的。

解决版本问题

1、卸载

先把torch-geometric、torch-sparse、torch-scatter、torch-cluster、 torch-spline-conv全部卸载了
pip uninstall torch-geometric torch-scatter torch-sparse torch-cluster torch-spline-conv

2、确定版本

conda list确定PyTorch的版本,我的是1.10在这里插入图片描述
nvcc -V确定cuda的版本,我的是11.3
在这里插入图片描述

3、确定下载地址

CPU版本的地址如下:torch-1.10.0+cpu的地址
https://pytorch-geometric.com/whl/torch-1.10.0+cpu.html
GPU版本的地址如下:torch-1.10.0+cuda11.3的地址
https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html

需要根据自己的版本更改网址!!!
网址能打开说明就ok,如果不存在那说明版本冲突


4、安装

我用是pip,通过-f指定链接

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.10.0+cu113.html

pip install torch-geometric

大功告成

在这里插入图片描述

标签:py,sparse,1.10,torch,报错,版本,geometric
来源: https://blog.csdn.net/qq_54478153/article/details/122683064