(cuda11.8 算力6.0-9.0) mamba_ssm-1.1.3 windows

作品简介

在Windows下安装 mamba_ssm 1.1.3,兼容算力低于7.0高于5.0的老旧GPU,同时支持7.0-9.0 等目前常见GPU

前期环境配置:

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
conda install packaging

下载后在这个文件目录下执行:

pip install mamba_ssm-1.1.3-cp310-cp310-win_amd64.whl

具体请参考本人CSDN博客及其系列博客:https://blog.csdn.net/yyywxk/article/details/140420538

注意 python torch 及 cuda 版本一定要与博客一致。

安装有问题加CSDN博客vx。

创作时间: