diff --git a/docker/trl-dev/Dockerfile b/docker/trl-dev/Dockerfile
index fdd470102b3..9a756a8821d 100644
--- a/docker/trl-dev/Dockerfile
+++ b/docker/trl-dev/Dockerfile
@@ -1,5 +1,5 @@
-FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime
+FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 RUN pip install --upgrade pip uv
 RUN uv pip install --system --no-cache "git+https://github.com/huggingface/trl.git#egg=trl[liger,peft,vlm]"
-RUN uv pip install --system hf_transfer kernels liger_kernel peft trackio
\ No newline at end of file
+RUN uv pip install --system kernels liger_kernel peft trackio
\ No newline at end of file
diff --git a/docker/trl/Dockerfile b/docker/trl/Dockerfile
index 1602fcca7e7..8b6e2842a38 100644
--- a/docker/trl/Dockerfile
+++ b/docker/trl/Dockerfile
@@ -1,3 +1,4 @@
-FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime
+FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
+RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 RUN pip install --upgrade pip uv
-RUN uv pip install --system trl[liger,peft,vlm] hf_transfer kernels trackio
\ No newline at end of file
+RUN uv pip install --system trl[liger,peft,vlm] kernels trackio
\ No newline at end of file