fix swift docker

This commit is contained in:
Jintao Huang
2025-07-31 15:48:56 +08:00
parent 63ea2244c4
commit c526ba7fb3

View File

@@ -87,7 +87,7 @@ if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \
cd /tmp && GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/NVIDIA/apex && \
cd apex && git checkout e13873debc4699d39c6861074b9a3b2a02327f92 && pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./ && \
cd / && rm -fr /tmp/apex && pip cache purge; \
pip install git+https://github.com/NVIDIA/Megatron-LM.git@core_r0.13.0; \
pip install git+https://github.com/NVIDIA/Megatron-LM.git@core_r0.12.0; \
fi
# install nvm and set node version to 18