# 正确删除整个环境
mamba deactivate || true
mamba env remove -n hf-train-cu118 || mamba remove -n hf-train-cu118 --all

# 仍然残留就手动清掉目录
rm -rf ~/.local/share/mamba/envs/hf-train-cu118

# 看看是否还在列表里
mamba env list


mamba env create -f hf-train-cu118.yaml --strict-channel-priority
mamba activate hf-train-cu118
python -m pip install --no-deps -r requirements-hf.txt

# deepspeed（选一）
# DS_BUILD_OPS=0 python -m pip install "deepspeed==0.14.*"
# 或
# mamba install -y cuda-toolkit=11.8 cmake ninja
# python -m pip install "deepspeed==0.14.*"