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.whl
pip install mamba_ssm-1.1.2-cp310-cp310-linux_x86_64.whl