torch-scatter import时候的问题version_cuda.so

本文介绍了如何解决因torch-scatter版本不匹配导致的错误,提供了检查torch和cuda版本的方法,并指导用户安装与当前torch版本相适配的torch-scatter等库。

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

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值