diff --git a/Dockerfile b/Dockerfile index 628ebfdf82c4..86905c87af2f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -54,7 +54,7 @@ RUN git clone https://github.com/NVIDIA/Megatron-LM.git && \ RUN git clone https://github.com/NVIDIA/apex.git && \ cd apex && \ git checkout 52e18c894223800cb611682dce27d88050edf1de && \ - pip install 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" ./ + 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" ./ RUN git clone https://github.com/NVIDIA/TransformerEngine.git && \ cd TransformerEngine && \ diff --git a/README.rst b/README.rst index fdd8ddb563f8..3dcf988f6ba2 100644 --- a/README.rst +++ b/README.rst @@ -280,7 +280,7 @@ To install Apex, run git clone https://github.com/NVIDIA/apex.git cd apex git checkout 52e18c894223800cb611682dce27d88050edf1de - pip install 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" ./ + 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.