[push-ci-image]

huggingface · Apr 28, 2024 · fdb9e25 · fdb9e25
1 parent aa62903
commit fdb9e25
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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