UltraLight-VM-UNet icon indicating copy to clipboard operation
UltraLight-VM-UNet copied to clipboard

mamba_ssm import Mamba

Open saccy6 opened this issue 10 months ago • 7 comments

你好 作者 非常感谢你对Mamba的贡献,但是这个ssm模块在哪里 我没有找到 该从哪里导入呢

saccy6 avatar Apr 11 '24 14:04 saccy6

Hi, you can find it in the official mamba code.

wurenkai avatar Apr 11 '24 14:04 wurenkai

感谢 看到相同的问题了 已解决

saccy6 avatar Apr 11 '24 14:04 saccy6

还有个问题 这个cuda必须11.7以上吗 我的是11.4可以跑吗

saccy6 avatar Apr 11 '24 14:04 saccy6

Currently mamba only supports Cuda11.7 or above, otherwise the environment installation will have problems and will not work properly.

wurenkai avatar Apr 11 '24 14:04 wurenkai

@wurenkai ImportError: libcudart.so.12: cannot open shared object file: No such file or directory How can this problem be solved? Thanks.

tiancpeng avatar Apr 25 '24 09:04 tiancpeng

@tiancpeng Hi, according to this issue, there should be a problem with the environment installation. Use "nvcc -V" to check your own Cuda version. The pytorch installation in the '0. Main Environments.' installation guide is based on Cuda 11.7, so please install pytorch according to your own Cuda version, and note that the Cuda version should be greater than or equal to 11.7.

wurenkai avatar Apr 25 '24 11:04 wurenkai

@wurenkai The Python interpreter could not find the shared library file named libcudart.so.12. libcudart.so.12 is part of CUDA Toolkit 12. I also checked my cuda version which is 11.7, configured my environment according to read.me, and ran python train.py still having the above problem, can't fix it!

tiancpeng avatar Apr 25 '24 14:04 tiancpeng