本文主要是介绍解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
介绍
Mamba: Linear-Time Sequence Modeling with Selective State Spaces
论文:https://arxiv.org/abs/2312.00752
Code:https://github.com/state-spaces/mamba 目前 5.3k Star
主要是为了解决pip install causal_conv1d
和mamba_ssm
老是安装出错。
包括但不限于各种 setup 不了,版本对不上…:
# 诸如
Building wheel for causal-conv1d (setup.py) ... error
error: subprocess-exited-with-error
所以直接建了个docker,但是还是有问题,最后按照 Mamba repo issue#55 的方法总算是装上了。
包含Mamba v1.1.1 的基础环境,Ubuntu20.04
,是以nvidia/cuda
官方11.7版本docker为基础构建的,具体有以下包:
cuda==11.7 (docker: nvidia/cuda)
pytorch==1.13
causal_conv1d==1.1.1
mamba_ssm==1.1.1
使用
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
# 这个docker的tag是1.1.1,请不要使用latest或者默认
测试
python
>>>import torch
>>>from mamba_ssm import Mamba
这篇关于解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!