Guide to Install on Windows

[siddagra]

You have to use WSL2!!

In WSL2:

First you have to make sure you install torch version that matches your nvcc version:

python3 -c 'import torch; print(torch.version.cuda)'
nvcc --version

You can match the versions by installing an older version of torch which uses a different cuda version:

pip3 install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu<major_version><minor_version>

then clone causal conv 1d and pip install:

git clone https://github.com/Dao-AILab/causal-conv1d.git
cd causal_conv1d
git checkout v1.0.2  # this is the highest compatible version allowed by Mamba
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip3 install .

then clone mamba and pip install:

https://github.com/state-spaces/mamba.git
cd mamba
pip3 install .

Please react with a thumbs-up/heart/rocket if it works so other people know this approach works as well.

[siddagra]

May be nice to add to README