From 729cc070fd72df6d15dec8a1b2fba36fd9b6ec7e Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Mon, 29 Apr 2024 17:05:03 +0200 Subject: [PATCH] nit --- docker/torch-light.dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index 03c2d1c4229822..8624fdf17e74ad 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -7,4 +7,5 @@ RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-cache-dir accelerate soundfile "fsspec>=2023.5.0,<2023.10.0" "transformers[sklearn,sentencepiece,vision,timm,testing]" RUN pip uninstall -y transformers && apt-get clean || apt-get -y --purge remove "*nvidia*" || apt-get autoremove || apt-get autoclean -RUN pip cache remove "nvidia-*" ||pip uninstall -y `pip freeze | grep "nvidia-*"` || pip uninstall -y `pip freeze | grep "triton-*"` || pip cache remove triton || true \ No newline at end of file +RUN pip cache remove "nvidia-*" || pip uninstall -y `pip freeze | grep "nvidia-*"` || pip uninstall -y `pip freeze | grep "triton-*"` || pip cache remove triton || true +RUN rm -rf /usr/local/lib/python3.10/site-packages/nvidia \ No newline at end of file