Import Error(_ZN2at4_ops10zeros_like4callERKNS_6TensorEN3c108optionalINS5_10ScalarTypeEEENS6_INS5_6LayoutEEENS6_INS5_6DeviceEEENS6_IbEENS6_INS5_12MemoryFormatEEE)

[AhJayzZ]

Hi, after I have successfully installed the mamba-ssm and causal-conv1d , the error shown

ImportError: /home/antony/MambaTrack/.venv/lib/python3.8/site-packages/selective_scan_cuda.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops10zeros_like4callERKNS_6TensorEN3c108optionalINS5_10ScalarTypeEEENS6_INS5_6LayoutEEENS6_INS5_6DeviceEEENS6_IbEENS6_INS5_12MemoryFormatEEE

[AhJayzZ]

This is the runtime and package version:

python 3.8.10

mamba-ssm==2.0.3
causal-conv1d==1.2.2
torch==2.3.0
nvidia-cuda-runtime-cu12==12.1.105
torchvision==0.18.0

[CYYJL]

Me too.
My package version:
python 3.10.14

mamba-ssm==2.0.2
causal-conv1d==1.2.0.post1
torch==2.1.1
torchvision==0.16.1
cuda==11.8

[AhJayzZ]

After I have change the CUDA to 11.8 and reduce the package version

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install causal_conv1d==1.1.1
pip install mamba-ssm==1.2.0.post1

It work!

[LaplaceSama]

Me three!