diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index c74b4594127d20..56c30c48cbb60b 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -6,9 +6,10 @@ RUN apt-get install -y g++ cmake ENV VIRTUAL_ENV=/usr/local RUN pip --no-cache-dir install uv RUN uv venv -RUN uv pip install --no-cache-dir -U pip setuptools accelerate soundfile +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 accelerate soundfile RUN uv pip install --no-cache-dir "transformers[sklearn,sentencepiece,vision,timm,testing]"