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