8000 GitHub - jiaowoguanren0615/Install_Mamba: This is an illustration for installing mamba_block & causal-conv1d.
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

jiaowoguanren0615/Install_Mamba

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
.gitignore
.gitignore
 
 
LICENSE
LICENSE
 
 
README.md
README.md
 
 

Repository files navigation

Install mamba_ssm & causal_conv1d

Two approaches (Docker Installation & Conda Installation)

Strongly recommend to create a new conda-environment!!!

Approach 1: Docker

<1> If you have not installed docker, run these commands step by step:

1. yum install -y yum-utils device-mapper-persistent-data lvm2

2. yum-config-manager --add-repo https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo

3. yum install -y docker-ce

4. systemctl start docker

5. systemctl enable docker

6. docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1

<2>If you have installed docker, just run the 6th command:

docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1

Approach 2: Conda environment

1. conda create -n mamba python=3.9

2. conda activate mamba

3. conda install cudatoolkit==11.8 -c nvidia

4. pip install torch==1.13 torchvision==0.14.0 torchaudio==0.13.0 --index-url https://download.pytorch.org/whl/cu117

5. conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc

6. conda install packaging

7. pip install setuptools==69.5.1

8. pip install causal-conv1d==1.2.1

9. pip install mamba-ssm

About

This is an illustration for installing mamba_block & causal-conv1d.

Resources

Readme

License

MIT license
Activity

Stars

1 star

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published
0