出现的问题: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.10nvcc -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