Open Barsuen opened 4 months ago
ImportError Traceback (most recent call last) Cell In[25], line 1 ----> 1 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
File ~/miniconda3/lib/python3.10/site-packages/mamba_ssm/init.py:3 1 version = "1.2.0.post1" ----> 3 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn 4 from mamba_ssm.modules.mamba_simple import Mamba 5 from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
File ~/miniconda3/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py:16 13 causal_conv1d_fn = None 14 causal_conv1d_cuda = None ---> 16 import selective_scan_cuda 19 class SelectiveScanFn(torch.autograd.Function): 21 @staticmethod 22 def forward(ctx, u, delta, A, B, C, D=None, z=None, delta_bias=None, delta_softplus=False, 23 return_last_state=False):
ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory
ImportError Traceback (most recent call last) Cell In[25], line 1 ----> 1 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
File ~/miniconda3/lib/python3.10/site-packages/mamba_ssm/init.py:3 1 version = "1.2.0.post1" ----> 3 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn 4 from mamba_ssm.modules.mamba_simple import Mamba 5 from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
File ~/miniconda3/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py:16 13 causal_conv1d_fn = None 14 causal_conv1d_cuda = None ---> 16 import selective_scan_cuda 19 class SelectiveScanFn(torch.autograd.Function): 21 @staticmethod 22 def forward(ctx, u, delta, A, B, C, D=None, z=None, delta_bias=None, delta_softplus=False, 23 return_last_state=False):
ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory