Broken Dependency in Docker

[kyleliang919]

Dockerfile:22
--------------------
87.47       RuntimeError: Error compiling objects for extension
87.47       [end of output]
87.47
87.47   note: This error originates from a subprocess, and is likely not a problem with pip.
87.48   ERROR: Failed building wheel for mamba-ssm
87.48 ERROR: Could not build wheels for mamba-ssm, which is required to install pyproject.toml-based projects
--------------------
ERROR: failed to solve: process "/bin/sh -c pip install --no-cache-dir datasets==2.20.0 triton==2.3.0 xformers tomli>=1.1.0 causal-conv1d>=1.2.0 mamba-ssm transformers" did not complete successfully: exit code: 1

My system spec:

+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 560.35.03              Driver Version: 560.35.03      CUDA Version: 12.6     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA H100 PCIe               On  |   00000000:01:00.0 Off |                    0 |
| N/A   30C    P0             44W /  350W |       1MiB /  81559MiB |      0%      Default |
|                                         |                        |             Disabled |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+

What I did:
Simply cloning the repo and ran docker build -t my-image .