OpenGVLab / VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding
https://arxiv.org/abs/2403.06977
Apache License 2.0
660 stars 47 forks source link

Layernorm: No such file or directory: 'ldconfig' #44

Closed ZXXaaaa closed 2 months ago

ZXXaaaa commented 2 months ago

When run hidden_states, residual = fused_add_norm_fn( hidden_states if residual is None else self.drop_path(hidden_states), self.norm.weight, self.norm.bias, residual=residual, prenorm=True, residual_in_fp32=self.residual_in_fp32, eps=self.norm.eps, ) then go into layernorm.py and run y, mean, rstd, residual_out = _layer_norm_fwd( x, weight, bias, eps, residual, residual_dtype=residual_dtype, is_rms_norm=is_rms_norm ) the function _layer_norm_fwd report the error: [Errno 2] No such file or directory: 'ldconfig' The full error message is as follows: [Errno 2] No such file or directory: 'ldconfig' File "/data2/ZYM/Net/networks/mamba/mamba_ssm/ops/triton/layernorm.py", line 155, in _layer_norm_fwd _layer_norm_fwd_1pass_kernel[(M,)]( File ""/data2/ZYM/Net/networks/networks/mamba/mamba_ssm/ops/triton/layernorm.py", line 411, in forward y, mean, rstd, residual_out = _layer_norm_fwd( File ""/data2/ZYM/Net/networks/networks/mamba/mamba_ssm/ops/triton/layernorm.py", line 478, in rms_norm_fn return LayerNormFn.apply(x, weight, bias, residual, eps, prenorm, residual_in_fp32, True) File ""/data2/ZYM/Net/networks/networks/VideoMamba.py", line 85, in forward hidden_states, residual = fused_add_norm_fn( File ""/data2/ZYM/Net/networks/networks/VideoMamba.py", line 346, in forward_features hidden_states, residual = layer( File ""/data2/ZYM/Net/networks/networks/VideoMamba.py", line 373, in forward x = self.forward_features(x, inference_params) File ""/data2/ZYM/Net/networks/train_with_videomamba.py", line 324, in frame_pred = videomamba_model(frame_input) FileNotFoundError: [Errno 2] No such file or directory: 'ldconfig' Has anyone encountered this error?

Andy1621 commented 2 months ago

Hi! It seems to be an error in your train_with_videomamba.py. Please check it.

ZXXaaaa commented 2 months ago

Thank you for your reply! I can run the same code on other servers, but when I run it on our lab server, I get this error. I guess there might be a problem with the environment, so I want to ask if anyone have encountered the same problem.