how to fix install error

[jiugexuan]

when i used install command,i get:

Building wheels for collected packages: causal-conv1d, mamba-ssm, buildtools, docopt
  error: subprocess-exited-with-error
  
  × python setup.py bdist_wheel did not run successfully.
  │ exit code: 1
  ╰─> See above for output.
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  Building wheel for causal-conv1d (setup.py) ... error
  ERROR: Failed building wheel for causal-conv1d
  Running setup.py clean for causal-conv1d
  error: subprocess-exited-with-error
  
  × python setup.py bdist_wheel did not run successfully.
  │ exit code: 1
  ╰─> See above for output.
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  Building wheel for mamba-ssm (setup.py) ... error
  ERROR: Failed building wheel for mamba-ssm
  Running setup.py clean for mamba-ssm
  Building wheel for buildtools (setup.py) ... done
  Created wheel for buildtools: filename=buildtools-1.0.6-py3-none-any.whl size=512341 sha256=a7d6709327166bee86a3c8f44fdfb6c0725c3e69cdb44c1284227bd880bd0c9b
  Stored in directory: /root/.cache/pip/wheels/90/e9/2a/625d99dffa430d0b4293d3d386f63e0eb8edeeb54f3f29d208
  Building wheel for docopt (setup.py) ... done
  Created wheel for docopt: filename=docopt-0.6.2-py2.py3-none-any.whl size=13704 sha256=d0220d812651f7ed34d19a7b5dcfd16b3ed5490c45444ebb926f50257b8e7298
  Stored in directory: /root/.cache/pip/wheels/fc/ab/d4/5da2067ac95b36618c629a5f93f809425700506f72c9732fac
Successfully built buildtools docopt
Failed to build causal-conv1d mamba-ssm
ERROR: ERROR: Failed to build installable wheels for some pyproject.toml based projects (causal-conv1d, mamba-ssm)

so can i firstly install mamba-ssm and then install the mambaVision?

[ahatamiz]

Hi @jiugexuan

I assume this is CUDA/torch incompatblity issue. Can you provide more details ? we recommend using newer version like CUDA 11.8+ and PyTorch 2.00+

[johnnynunez]

export MAX_JOBS=$(nproc)
export CAUSAL_CONV1D_FORCE_BUILD="TRUE"
export CAUSAL_CONV1D_SKIP_CUDA_BUILD="FALSE"