diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 72853d4ca4..80f44f6294 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -71,6 +71,9 @@ RUN python3 -m pip install --no-cache-dir g2p-en # For Some bitsandbytes tests RUN python3 -m pip install --no-cache-dir einops +# `kernels` may give different outputs (within 1e-5 range) even with the same model (weights) and the same inputs +RUN python3 -m pip uninstall -y kernels + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop diff --git a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile index 45aa89fefb..746026d7ec 100644 --- a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile +++ b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile @@ -45,6 +45,9 @@ RUN python3 -m pip uninstall -y deepspeed # TODO: Find out why test fail. RUN DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 python3 -m pip install deepspeed --global-option="build_ext" --global-option="-j8" --no-cache -v --disable-pip-version-check 2>&1 +# `kernels` may give different outputs (within 1e-5 range) even with the same model (weights) and the same inputs +RUN python3 -m pip uninstall -y kernels + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop diff --git a/docker/transformers-pytorch-deepspeed-nightly-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-nightly-gpu/Dockerfile index 9daa27c06e..d71b4ad334 100644 --- a/docker/transformers-pytorch-deepspeed-nightly-gpu/Dockerfile +++ b/docker/transformers-pytorch-deepspeed-nightly-gpu/Dockerfile @@ -57,6 +57,9 @@ RUN python3 -m pip uninstall -y deepspeed #RUN git clone https://github.com/pytorch/TensorRT.git #RUN cd TensorRT/py && python3 setup.py install --fx-only +# `kernels` may give different outputs (within 1e-5 range) even with the same model (weights) and the same inputs +RUN python3 -m pip uninstall -y kernels + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop diff --git a/docker/transformers-pytorch-gpu/Dockerfile b/docker/transformers-pytorch-gpu/Dockerfile index 7ef08e4d83..ee7f8a8403 100644 --- a/docker/transformers-pytorch-gpu/Dockerfile +++ b/docker/transformers-pytorch-gpu/Dockerfile @@ -28,6 +28,9 @@ RUN python3 -m pip uninstall -y tensorflow flax RUN python3 -m pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git pytesseract RUN python3 -m pip install -U "itsdangerous<2.1.0" +# `kernels` may give different outputs (within 1e-5 range) even with the same model (weights) and the same inputs +RUN python3 -m pip uninstall -y kernels + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop diff --git a/docker/transformers-quantization-latest-gpu/Dockerfile b/docker/transformers-quantization-latest-gpu/Dockerfile index 504ed9704c..c860dabd6a 100755 --- a/docker/transformers-quantization-latest-gpu/Dockerfile +++ b/docker/transformers-quantization-latest-gpu/Dockerfile @@ -90,6 +90,9 @@ RUN python3 -m pip install --no-cache-dir "auto-round>=0.5.0" # Add transformers in editable mode RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch] +# `kernels` may give different outputs (within 1e-5 range) even with the same model (weights) and the same inputs +RUN python3 -m pip uninstall -y kernels + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop