search

hustvl / Vim

Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Apache License 2.0
2.55k stars 159 forks source link

ImportError: libcudart.so.12: cannot open shared object file: No such file or directory #68

Closed Liuhp133 closed 2 months ago

Liuhp133 commented 2 months ago

Traceback (most recent call last): File "/home/omnisky/sdf/ljx/Vim/vim/main.py", line 28, in import models_mamba File "/home/omnisky/sdf/ljx/Vim/vim/models_mamba.py", line 20, in from mamba_ssm.modules.mamba_simple import Mamba File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py", line 9, in from causal_conv1d import causal_conv1d_fn File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/init.py", line 3, in from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/causal_conv1d_interface.py", line 7, in import causal_conv1d_cuda ImportError: libcudart.so.12: cannot open shared object file: No such file or directory Traceback (most recent call last): File "/home/omnisky/sdf/ljx/Vim/vim/main.py", line 28, in import models_mamba File "/home/omnisky/sdf/ljx/Vim/vim/models_mamba.py", line 20, in from mamba_ssm.modules.mamba_simple import Mamba File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py", line 9, in from causal_conv1d import causal_conv1d_fn File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/init.py", line 3, in from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/causal_conv1d_interface.py", line 7, in import causal_conv1d_cuda ImportError: libcudart.so.12: cannot open shared object file: No such file or directory Traceback (most recent call last): File "/home/omnisky/sdf/ljx/Vim/vim/main.py", line 28, in import models_mamba File "/home/omnisky/sdf/ljx/Vim/vim/models_mamba.py", line 20, in from mamba_ssm.modules.mamba_simple import Mamba File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/home/omnisky/sdf/ljx/Vim/mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py", line 9, in from causal_conv1d import causal_conv1d_fn File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/init.py", line 3, in from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/causal_conv1d/causal_conv1d_interface.py", line 7, in import causal_conv1d_cuda ImportError: libcudart.so.12: cannot open shared object file: No such file or directory [2024-04-19 14:55:55,003] torch.distributed.elastic.multiprocessing.api: [ERROR] failed (exitcode: 1) local_rank: 0 (pid: 13126) of binary: /home/omnisky/anaconda3/envs/vim/bin/python Traceback (most recent call last): File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/runpy.py", line 86, in _run_code exec(code, run_globals) File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/launch.py", line 196, in main() File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/launch.py", line 192, in main launch(args) File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/launch.py", line 177, in launch run(args) File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/run.py", line 797, in run elastic_launch( File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 134, in call return launch_agent(self._config, self._entrypoint, list(args)) File "/home/omnisky/anaconda3/envs/vim/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 264, in launch_agent raise ChildFailedError( torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

zyszxhy commented 1 month ago

Hello, have you solved this problem yet?