Dao-AILab / causal-conv1d

Causal depthwise conv1d in CUDA, with a PyTorch interface
BSD 3-Clause "New" or "Revised" License
325 stars 60 forks source link

ImportError #35

Open qinsiyuan-cool opened 1 month ago

qinsiyuan-cool commented 1 month ago

my env: Python 3.10.13 torch version 2.4.1 Cuda compilation tools, release 12.6, V12.6.68 Build cuda_12.6.r12.6/compiler.34714021_0 I install "causal_conv1d-1.4.0+cu122torch2.4cxx11abiTRUE-cp310-cp310-linux_x86_64.whl" successfully by pip but when i run my program,it give an Error"ImportError: /data/conda_envs/.../lib/python3.10/site-packages/causal_conv1d_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c107WarningC1ESt7variantIJNS0_11UserWarningENS0_18DeprecationWarningEEERKNS_14SourceLocationENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEEb "

sadimanna commented 1 month ago

I was having the same issue right now. I was also getting the error when using torch-2.4.1 compiled with cuda12.1. I think this happened sue to using a pre-built wheel on a system with a different cuda version. I simply uninstalled the version installed using wheel. And then simply executed the command

python3 -m pip install causal_conv1d>=1.4.0 mamba_ssm>=2.2.2

After that i got no error when importing.