发布时间: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
webpack--》webpack底层深入讲解,从初识到精通,真正实现从0到1的过程
Pytorch学习——自动求导机制requires_grad()
在TensorRT中使用动态形状(Dynamic Shapes)
你或许也想拥有专属于自己的AI模型文件格式(推理部署篇)-(7)
SpingCloud集成zookeeper实现服务注册并访问
【详解】Springboot热部署devtools的使用,设置刷新时间
inject 不生效?!依赖注入背后的实现原理和运行逻辑是怎样的?
基于servlet+jsp+mysql实现的工资管理系统【源码+数据库】