diff --git a/Dockerfile b/Dockerfile index 094c8fff408e..10aa6c01f3a0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -72,7 +72,7 @@ RUN git clone https://github.com/NVIDIA/Megatron-LM.git && \ # Apex bugfix for PyTorch 23.11 container: https://github.com/NVIDIA/apex/pull/1760 RUN git clone https://github.com/NVIDIA/apex.git && \ cd apex && \ - git checkout c07a4cf67102b9cd3f97d1ba36690f985bae4227 && \ + git checkout 52e18c894223800cb611682dce27d88050edf1de && \ pip install -v --no-build-isolation --disable-pip-version-check --no-cache-dir --config-settings "--build-option=--cpp_ext --cuda_ext --fast_layer_norm --distributed_adam --deprecated_fused_adam" ./ # Transformer Engine 1.2.0 diff --git a/README.rst b/README.rst index 78396d80dc45..73e453d775f3 100644 --- a/README.rst +++ b/README.rst @@ -326,7 +326,7 @@ To install Apex, run git clone https://github.com/NVIDIA/apex.git cd apex - git checkout c07a4cf67102b9cd3f97d1ba36690f985bae4227 + git checkout 52e18c894223800cb611682dce27d88050edf1de pip install -v --no-build-isolation --disable-pip-version-check --no-cache-dir --config-settings "--build-option=--cpp_ext --cuda_ext --fast_layer_norm --distributed_adam --deprecated_fused_adam" ./ It is highly recommended to use the NVIDIA PyTorch or NeMo container if having issues installing Apex or any other dependencies.