From fdb9e2552f036049145df88a99750cc3a854a694 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Sun, 28 Apr 2024 15:43:38 +0200 Subject: [PATCH] [push-ci-image] --- docker/torch-light.dockerfile | 4 +++- docker/torch-tf-light.dockerfile | 5 +++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index 95c8a7c70d2112..c879e23c71192c 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -9,7 +9,9 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools RUN uv pip install --no-cache-dir "fsspec>=2023.5.0,<2023.10.0" RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu -RUN uv pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm,testing]" +RUN pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm,testing]" + +# sklearn,sentencepiece,vision,timm,testing fucking install nividia-ci RUN pip uninstall -y transformers diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile index c5db0feb08974b..9c28ef6dd600f5 100644 --- a/docker/torch-tf-light.dockerfile +++ b/docker/torch-tf-light.dockerfile @@ -13,12 +13,13 @@ RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url ht RUN uv pip install --no-cache-dir pypi-kenlm protobuf==3.20.3 accelerate tensorflow_probability pytest pytest-xdist parameterized RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,sentencepiece,vision]" + RUN pip uninstall -y transformers RUN apt-get clean && rm -rf /var/lib/apt/lists/* RUN pip cache remove "nvidia-*" RUN pip uninstall -y `pip freeze | grep "nvidia-*"` || true RUN pip cache remove triton -RUN apt-get --purge remove "*nvidia*" -RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*" +RUN apt-get --purge remove "*nvidia*" || true +RUN apt-get --purge remove "*cublas*" "cuda*" "nsight*" || true RUN apt-get autoremove RUN apt-get autoclean \ No newline at end of file