由于现在mamba大火,所以将mamba_ssm在windows中的环境配置进行介绍,如果你想在windows中进行开发,那么这是你最好的选择!!
安装步骤
1.anaconda下载
官网下载:https://www.anaconda.com/download
下载好后:
接下来点击进行安装,对于anaconda的安装,网上教程很多,请自行安装
下载CUDA
1.win+r打开cmd,输入nvidia-smi查看自己的cuda version版本。
2.下载cuda
下载地址:CUDA Toolkit Archive:https://developer.nvidia.com/cuda-toolkit-archive
我下载的是cuda11.8版本,因为目前我尝试的mamba_ssm环境配置中,只有cuda11.8不报错!!!!!!!!!!!!
下载到自己电脑上,然后点击安装,然后一路next就行
3.cudnn安装
下载地址:https://developer.nvidia.com/rdp/cudnn-archive
下载下来将下面三个文件复制到cuda对应文件夹内
mamba_ssm下载安装
1.打开anaconda中的anaconda prompt
2.创建虚拟环境mamba,python版本设置为3.10.13,!!!!!别的python版本不行!!!!!!!:
conda create -n mamba python=3.10.13
3.激活创建的mamba虚拟环境:
conda activate mamba
4.下载torch和对应的torchvision和torchaudio
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
5.下载triton,这个包很重要,没有这个包,mamba_ssm包没办法下载,但是triton是专门服务于ubuntu的,没有windows版本,所以找了很久,终于找到了triton的windows环境下使用的包:
trion下载到本地后,需要在你下载的位置,进行安装:
6.安装causal_conv1d,因为causal_conv1d和mamba是一起使用的,所以需要先安装causal_conv1d,目前只有下载到本地,可以安装成功,而且这个文件是经过编译的,可以直接下载安装,网上其他方法都会报错,因为环境不匹配!!!!!!!注意:
**上图是我下载到我电脑上的,编译好的,可以直接下载到本地使用,
7.安装mamba_ssm
同样的,mamba_ssm包只有下载到本地,再进行安装,才可以不报错,网上其他都有问题!!!!
下载到本地:
下载安装:
这样mamba_ssm就安装好了,可以去搞创新点和发论文楼!!!!!!!!!!!