torch-scatter import时候的问题

torch_scatter/_version_cuda.so: undefined symbol: ZN3c106detail12infer_schema20make_function_schemaENS_8ArrayRefINS1_11ArgumentDefEEES4

遇到了以上的报错,该报错是scatter,从报错的文件名不难猜测出事关于版本的问题,这个问题的具体原因也便是因为你主机上的与torch相关的文件版本不适配,所以我们需要将有关torch的版本调为统一。
pytorch网站,根据该网站,我们将torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric几个库更新为与我们电脑上torch版本一致。
查询torch,cuda版本代码:

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-{version}+cu{version}.html

在这里插入图片描述
问题解决。

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐