hailin
/
train_env_prepare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-08 17:45:22 +08:00
parent 5868300787
commit 1e9d6ffd7d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
README.txt
View File

@ -1,9 +1,9 @@
mamba env create -f env.yaml --strict-channel-priority
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.*"
# 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.*"
# mamba install -y cuda-toolkit=11.8 cmake ninja
# python -m pip install "deepspeed==0.14.*"