undefined symbol: _ZN3c104cuda9SetDeviceEab

详细问题

File "/home/wanghao/ultralytics-main/ultralytics/nn/__init__.py", line 3, in <module>from .tasks import (File "/home/wanghao/ultralytics-main/ultralytics/nn/tasks.py", line 14, in <module>from ultralytics.nn.modules import (File "/home/wanghao/ultralytics-main/ultralytics/nn/modules/__init__.py", line 106, in <module>from .mambablock import MambaBlockFile "/home/wanghao/ultralytics-main/ultralytics/nn/modules/mambablock.py", line 3, in <module>from mamba_ssm import Mamba2File "/home/wanghao/miniconda3/envs/mamba2/lib/python3.12/site-packages/mamba_ssm/__init__.py", line 3, in <module>from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fnFile "/home/wanghao/miniconda3/envs/mamba2/lib/python3.12/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 20, in <module>import selective_scan_cuda
ImportError: /home/wanghao/miniconda3/envs/mamba2/lib/python3.12/site-packages/selective_scan_cuda.cpython-312-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda9SetDeviceEab

解决方法

卸载mamba-ssm，执行命令：

pip uninstall mamba-ssm

获取github代码，链接如下：

https://github.com/state-spaces/mamba

然后进入目录，执行安装命令：

cd mamba-main
python setup.py install