发布时间:2022-09-13 15:30
用到torch_geometric
。但是写好代码后发现会出现报错:
ModuleNotFoundError: No module named 'torch_sparse'
可以说这些包是真的难装,我一直会出现ERROR: Failed building wheel for torch-sparse
这个问题,按我的理解来讲,这个问题有时候极有可能是包之间的版本问题引起的。然后在网上查了一下,果然是这样的。
先把torch-geometric、torch-sparse、torch-scatter、torch-cluster、 torch-spline-conv全部卸载了
pip uninstall torch-geometric torch-scatter torch-sparse torch-cluster torch-spline-conv
conda list
确定PyTorch的版本,我的是1.10
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,如果不存在那说明版本冲突
我用是pip,通过-f指定链接
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.10.0+cpu.html
pip install torch-geometric