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