当前位置:   article > 正文

安装torch_scatter

安装torch_scatter

ModuleNotFoundError: No module named 'torch_scatter'
 

一、找对应版本

需要看python版本,pytorch版本,cuda组件版本。

终端输入:

conda list

看一下,我安装的pytorch为1.12.1+cu113

也就是1.12.1的pytorch和11.3的cuda组件,python是3.8

对应去网站查找下载:

https://pytorch-geometric.com/whl/index.html

找到

点进去找对应python(cp38)的ubuntu版本的scatter

下载好后直接pip安装

pip install path../torch_scatter-2.1.0+pt112cu113-cp38-cp38-linux_x86_64.whl

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/458793
推荐阅读
相关标签
  

闽ICP备14008679号