This commit is contained in:
parent
5868300787
commit
1e9d6ffd7d
|
|
@ -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
|
mamba activate hf-train-cu118
|
||||||
python -m pip install --no-deps -r requirements-hf.txt
|
python -m pip install --no-deps -r requirements-hf.txt
|
||||||
|
|
||||||
# deepspeed(选一)
|
# 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
|
# mamba install -y cuda-toolkit=11.8 cmake ninja
|
||||||
python -m pip install "deepspeed==0.14.*"
|
# python -m pip install "deepspeed==0.14.*"
|
||||||
Loading…
Reference in New Issue