No module named ‘pytorch-geometric’/‘torch_scatte’/ ‘torch_scatter.scatter_cuda‘

ie_czg

于 2022-06-20 10:06:43 发布

阅读量519

点赞数

CC 4.0 BY-SA版权

分类专栏：我的学习记录文章标签： pytorch 深度学习

本文链接：https://blog.youkuaiyun.com/C20162430202/article/details/125366966

我的学习记录专栏收录该内容

4 篇文章

订阅专栏

当遇到'pytorch-geometric'、'torch_scatte'、'torch_scatter.scatter_cuda'找不到模块的错误时，问题可能出在PyTorch、CUDA版本不匹配。为了解决这个问题，需要按照PyTorch-Geometric官网指示安装与当前环境兼容的版本。首先，检查PyTorch和CUDA的版本，然后参照官方文档选择正确的安装命令。使用conda命令进行安装可以确保版本对应，从而解决导入错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

No module named ‘pytorch-geometric’/'torch_scatte’/ 'torch_scatter.scatter_cuda’即使安装过pytorch-geometric依然报错
原因：pytorch-geometric、torch、cuda版本不对应
解决：按照pytorch-geometric官网安装对应版本的pytorch-geometric

首先查看torch、cuda版本

python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.version.cuda)"

![在这里插入图片描述](https://img-blog.csdnimg.cn/969080055e7e486984c66f695e0ca39b.png

然后在官网https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html中选择正确的安装命令。
在这里插入图片描述 conda命令安装

conda install pyg -c pyg

安装成功
在这里插入图片描述