近期在配置manba vision的环境,其需要下载一个要编译的包mamba-ssm。编译过程耗时很长且容易因为CUDA版本不一致而编译失败。
在AUTODL平台的pip默认是从阿里云下载,至于从github上下载release的.whl文件无法做到,因此采用办法:先从Releases · state-spaces/mamba中下载对应的.whl文件。
这里我选择的是mamba_ssm-2.2.4+cu12torch2.5cxx11abiFALSE-cp312-cp312-linux_x86_64
也即cu12版本,torch 2.5,python3.12
然后再把.whl文件上传到AUTODL平台,再用pip install ****.whl文件来安装。
