state-spaces / mamba

Mamba SSM architecture
Apache License 2.0
13.12k stars 1.11k forks source link

How does the code run on the Macbook #546

Open 524125153 opened 2 months ago

524125153 commented 2 months ago

My current device is only Macbook, its corresponding chip is M1 chip, and there is no cuda, in this case, I can not run this code?

Anri-Lombard commented 2 months ago

Exactly right. You need to have Cuda to use mamba; I've had the same experience as you a while back (luckily I use our university HPC cluster, so do have the needed GPU). You could run it on Google Colab or Kaggle if you want, just watch out since you'll have limited GPU time.

josmithiii commented 2 months ago

You can run mamba-minimal on a Mac: https://github.com/johnma2006/mamba-minimal.git