cuda11.4 ImportError: libcudart.so.9.2: cannot open shared object file: No such file or directory

vidt配环境过程中执行完

cd ./ops
sh ./make.sh
# unit test (should see all checking is True)
python test.py

后tset.py报错

Traceback (most recent call last):
  File "test.py", line 18, in <module>
    from functions.ms_deform_attn_func import MSDeformAttnFunction, ms_deform_attn_core_pytorch
  File "/home/cse305/code/vidt-main/ops/functions/__init__.py", line 9, in <module>
    from .ms_deform_attn_func import MSDeformAttnFunction
  File "/home/cse305/code/vidt-main/ops/functions/ms_deform_attn_func.py", line 18, in <module>
    import MultiScaleDeformableAttention as MSDA
ImportError: libcudart.so.9.2: cannot open shared object file: No such file or directory

可能原因

本机使用cuda版本为11.4,不知道为什么这里import MSDA 的时候好像在找cuda 9.2

解决方法

先把ops中原来编译的build文件夹删掉,在终端中执行

export CUDA_HOME=/usr/local/cuda-11.4 #具体版本查看自己电脑的路径

然后重新编译一次,就不会报错了(太感人了,不用重新装低版本cuda)

p.s.其他问题指路https://github.com/fundamentalvision/Deformable-DETR/issues

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值