可能有用的操作 创建conda环境,在conda环境安装 pip install torch pip install packaging 可能需要学术加速(?不清楚) 现在的安装方法: 对于mamba-ssm 根据cuda、torch、python版本在mamba-ssm官方releases下载对应的包 然后pip install安装,比如 pip install mamba_ssm-2.2.2+cu118torch2.1cxx11abiFALSE-cp310-cp310...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 2、直接下载工程文件,...
运行test_mamba_module.py和test_selective_scan.py验证mamba-ssm,如果报错,没关系,直接找到你要运行的那个网络的文件,看看是否能正常输出,如果可以,就可以用了(有些时候作者修改了mamba_simple.py等文件,只要能保证你要用到的网络可以正常运行就行) 举例:在vivim.py里加入 if __name__ == "__main__" : ...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像 DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general 代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 2、直接下载工程文...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像 DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general 代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 ...