A well-built wheel is used to support the Nvidia 10 series graphics card. It was built under the Python 3.10,PyTorch 2.1 and CUDA 11.8 environment. For detailed construction details, please refer to the link.
pip install causal_conv1d-1.1.1-cp310-cp310-linux_x86_64.whlpip install mamba_ssm-1.1.2-cp310-cp310-linux_x86_64.whlFor detailed build process, please refer to build_step.pdf.