diff options
author | 2018-04-20 14:25:57 -0700 | |
---|---|---|
committer | 2018-04-20 14:28:22 -0700 | |
commit | 9fc5bacba49eb31c7d536963879ccc62ecfbaf76 (patch) | |
tree | 40988a0ee0f8a0208bb57c589c77dcfe7fcaa197 /tensorflow/tools/ci_build/install | |
parent | d82d04f15992e224743f29aa75134ed04aa064a7 (diff) |
Pin rbe-debian8-tf container tp a newer base image
- Also improve how numpy is installed (not compiling from source) for containers based on other distros than Ubuntu14.04
PiperOrigin-RevId: 193722848
Diffstat (limited to 'tensorflow/tools/ci_build/install')
-rwxr-xr-x | tensorflow/tools/ci_build/install/install_pip_packages.sh | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/tools/ci_build/install/install_pip_packages.sh b/tensorflow/tools/ci_build/install/install_pip_packages.sh index 9644277fab..5aaf544afd 100755 --- a/tensorflow/tools/ci_build/install/install_pip_packages.sh +++ b/tensorflow/tools/ci_build/install/install_pip_packages.sh @@ -65,8 +65,13 @@ rm -rf /usr/lib/python3/dist-packages/six* # numpy needs to be installed from source to fix segfaults. See: # https://github.com/tensorflow/tensorflow/issues/6968 # This workaround isn't needed for Ubuntu 16.04 or later. -pip2 install --no-binary=:all: --upgrade numpy==1.12.0 -pip3 install --no-binary=:all: --upgrade numpy==1.12.0 +if $(cat /etc/*-release | grep -q 14.04); then + pip2 install --no-binary=:all: --upgrade numpy==1.12.0 + pip3 install --no-binary=:all: --upgrade numpy==1.12.0 +else + pip2 install --upgrade numpy==1.12.0 + pip3 install --upgrade numpy==1.12.0 +fi pip2 install scipy==0.18.1 pip3 install scipy==0.18.1 |