目录
一、Win系统安装详细教程
安装系统要求:cuda11.8、python3.10
1、准备
根据下面的步骤,创建虚拟环境(如果已经有对应的环境,则可以跳过):
conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
其中,your_env_name是自己的虚拟环境名字。
注意:安装前没有packaging的需要先安装:
conda install packaging
上述工作准备好后,按下面步骤安装causal_conv1d和mamba_ssm。
2、安装Triton
Triton是安装mamba的依赖,需要先安装。