Repository navigation
No module named 'v2dmamba_scan.scan_ops' #20
Description
Hi,
I used your Docker to run ../TwoDMamba-main/TwoDVMamba/classification/main.py, but I got the following Traceback:
Traceback (most recent call last): File "/workspace/Code/TwoDMamba-main/TwoDVMamba/classification/models/vmamba.py", line 28, in <module> from .csms6s import selective_scan_fn, selective_scan_flop_jit File "/workspace/Code/TwoDMamba-main/TwoDVMamba/classification/models/csms6s.py", line 38, in <module> raise ImportError("Can not import v2dmamba_scan.") ImportError: Can not import v2dmamba_scan
I have already built the 2DMamba CUDA kernel.
I would like to know whether this is a compilation issue or a missing file issue.
Activity
jingweizhang-xyz commentedon May 28, 2026
Did you build the kernel inside the docker? Do you see the compiled .so file? You can refer to https://github.com/cvlab-stonybrook/LBMamba#build-lbmamba-cuda-kernel on how to see if the build is correct.