diff --git a/docker/torch-jax-light.dockerfile b/docker/torch-jax-light.dockerfile
index a81d3dc0e8a19c..9376ac4e975f8a 100644
--- a/docker/torch-jax-light.dockerfile
+++ b/docker/torch-jax-light.dockerfile
@@ -9,7 +9,7 @@ RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools accelerate
 RUN uv pip install --no-cache-dir "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist
 
-RUN uv pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu
+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,flax,torch,testing,sentencepiece,torch-speech,vision]"
 
 
diff --git a/docker/torch-tf-light.dockerfile b/docker/torch-tf-light.dockerfile
index 6acf843e7be12e..17534bbd747316 100644
--- a/docker/torch-tf-light.dockerfile
+++ b/docker/torch-tf-light.dockerfile
@@ -9,7 +9,7 @@ RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
 
 
-RUN uv pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu
 RUN uv pip install --no-cache-dir pypi-kenlm
 RUN uv pip install --no-cache-dir "transformers[sklearn,tf-cpu,torch,testing,sentencepiece,torch-speech,vision]"
 RUN uv pip install --no-cache-dir -U tensorflow_probability accelerate