1、创建Anaconda环境
conda create -n your_env_name python=3.10
conda activate your_env_name
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install seaborn thop timm einops
2、安装causal_conv1d和mamba_ssm库
直接用wh