Only upgrade pyarrow on CPU image

Author: djherbis

Dockerfile.tmpl

@@ -114,7 +114,7 @@ RUN pip install spacy && \
 
 # b/232247930: uninstall pyarrow to avoid double installation with the GPU specific version.
 RUN pip uninstall -y pyarrow && \
-    mamba install -y "pyarrow>=14.0.1" cudf cuml && \
+    mamba install -y cudf cuml && \
     /tmp/clean-layer.sh
 
 # TODO: b/296444923 - Resolve pandas dependency another way
@@ -481,7 +481,7 @@ RUN pip install bleach \
         webencodings \
         widgetsnbextension \
         # Require pyarrow newer than https://github.com/advisories/GHSA-5wvp-7f3h-6wmm
-        "pyarrow>=14.0.1" \
+        {{ if eq .Accelerator "gpu" }} pyarrow {{ else }} "pyarrow>=14.0.1" {{ end }} \
         feather-format \
         fastai