rm docker deps

google · Mar 21, 2024 · 1b4c3d7 · 1b4c3d7
1 parent 523b412
commit 1b4c3d7
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/docker/xla.Dockerfile b/docker/xla.Dockerfile
@@ -28,9 +28,7 @@ RUN apt-get install -y --no-install-recommends git
 # Install libraries.
 ENV PIP_ROOT_USER_ACTION=ignore
 RUN python3 -m pip install --upgrade pip
-RUN pip install fairscale==0.4.13
 RUN pip install numpy==1.24.4
-RUN pip install immutabledict==4.1.0
 RUN pip install sentencepiece==0.1.99
 
 # Install from source.

diff --git a/docker/xla_gpu.Dockerfile b/docker/xla_gpu.Dockerfile
@@ -31,9 +31,7 @@ RUN python3 -m pip install --upgrade pip
 RUN pip uninstall -y torch
 RUN pip install torch==2.1.1
 RUN pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/11.8/torch_xla-2.2.0rc1-cp38-cp38-linux_x86_64.whl
-RUN pip install fairscale==0.4.13
 RUN pip install numpy==1.24.4
-RUN pip install immutabledict==4.1.0
 RUN pip install sentencepiece==0.1.99
 
 # Install from source.