Mamba环境配置教程
一、前言
直接使用 pip install causal-conv1d 和 pip install mamba-ssm安装Mamba百分之九十九的情况下,大家都会遇到各种奇葩的Bug,例如下载网络请求超时、CUDA版本不匹配、C++编译失败等等。
使用离线安装方法可以有效地结果这个难题
二、方法
- 创建conda环境
conda create -n mamba python=3.10
- 进入创建的环境
conda activate mamba
- 安装torch以及相应的 torchvison、torchaudio。这里选择2.2版本,从Pytorch官网选择安装命令
conda install pytorch==