ImportError: .xx/elsa/elsa_ext.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBas

具体报错如下:

ImportError:./ELSA/det/mmdet/models/backbones/elsa/elsa_ext.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v

方案一
补充安装如下:

pip install torch-cluster
pip install torch-scatter
pip install torch-sparse

方案二
检差cuda的版本和pytorch 的cuda版本是否一致

# 查看cuda的版本
nvcc -V 		
			
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
# 查看pytorch 的cuda版本
import torch
torch.version.cuda	

'10.2'

不一致就安装
conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch

方案三
重新生成.so文件

cd ./cls/models/elsa
python setup.py install
mv build/lib*/* .
cp *.so ../../../det/mmdet/models/backbones/elsa/
cp *.so ../../../seg/mmseg/models/backbones/elsa/
cd ../../../

应该方案三work

参考链接1
参考链接2
参考链接3

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值