ImportError: libcudart.so.12: cannot open shared object file: No such file or directory #706
Description
Hi iam using ubuntu 20.04 WSL, python 3.10.6 and torch 2.0.1+cu118 causal_conv1d 1.2.0 and i get the following error when just trying to import mamba_ssm.
(umamba) bounty@LAPTOP-TISOQDL3:/mnt/c/UNM/Year 3/FYP/Umambaliver/U-Mamba/umamba$ python
Python 3.10.16 (main, Dec 11 2024, 16:24:50) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import mamba_ssm
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/bounty/miniconda3/envs/umamba/lib/python3.10/site-packages/mamba_ssm/__init__.py", line 3, in <module>
from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
File "/home/bounty/miniconda3/envs/umamba/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 18, in <module>
import selective_scan_cuda
ImportError: libcudart.so.12: cannot open shared object file: No such file or directory
>>> exit()
(umamba) bounty@LAPTOP-TISOQDL3:/mnt/c/UNM/Year 3/FYP/Umambaliver/U-Mamba/umamba$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0
(umamba) bounty@LAPTOP-TISOQDL3:/mnt/c/UNM/Year 3/FYP/Umambaliver/U-Mamba/umamba$
What is the problem exactly? Does it require CUDA 12x? I have the 11.8 installed because its a requisite for https://github.com/bowang-lab/U-Mamba?tab=readme-ov-file#installation. Any insight would be appreciated. Thanks!