jaxlib build improvements (jax-ml#1742)

build/Dockerfile

@@ -2,6 +2,8 @@ FROM gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010
 LABEL maintainer "Matt Johnson <[email protected]>"
 
 WORKDIR /
+RUN apt-get update
+RUN apt-get install libffi-dev
 RUN git clone --branch v1.2.14 https://github.com/pyenv/pyenv.git /pyenv
 ENV PYENV_ROOT /pyenv
 RUN /pyenv/bin/pyenv install 2.7.15

build/build_jaxlib_wheels.sh

@@ -13,7 +13,7 @@ for PYTHON_VERSION in $PYTHON_VERSIONS
 do
   mkdir -p dist/nocuda/
   docker run -it --tmpfs /build:exec --rm -v $(pwd)/dist:/dist jaxbuild $PYTHON_VERSION nocuda
-  mv dist/*.whl dist/nocuda/
+  mv -f dist/*.whl dist/nocuda/
 done
 
 # build the cuda linux packages, tagging with linux_x86_64
@@ -26,7 +26,7 @@ do
     do
       mkdir -p dist/${CUDA_VARIANT}${CUDA_VERSION//.}
       docker run -it --tmpfs /build:exec --rm -v $(pwd)/dist:/dist jaxbuild $PYTHON_VERSION $CUDA_VARIANT
-      mv dist/*.whl dist/${CUDA_VARIANT}${CUDA_VERSION//.}/
+      mv -f dist/*.whl dist/${CUDA_VARIANT}${CUDA_VERSION//.}/
     done
   done
 done