experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #46691 from ROCmSoftwarePlatform:google_upstream_rocm_container_pip_install_fix

Browse Source 
PiperOrigin-RevId: 357723262
Change-Id: I0463258f44fc97781b100bb684ba370cc74e2e76
This commit is contained in:
TensorFlower Gardener 2021-02-16 08:29:36 -08:00
commit 1aa2aafaad
1 changed files with 0 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tensorflow/tools/ci_build/install/install_pip_packages.sh
View File

@ -19,7 +19,6 @@ set -e
# Get the latest version of pip so it recognize manylinux2010
wget https://bootstrap.pypa.io/get-pip.py
python3.6 get-pip.py
python get-pip.py
rm -f get-pip.py
# Install pip packages from whl files to avoid the time-consuming process of
@ -27,41 +26,31 @@ rm -f get-pip.py
# Pin wheel==0.31.1 to work around issue
# https://github.com/pypa/auditwheel/issues/102
pip2 install wheel==0.31.1
pip3 install wheel==0.31.1
# Install last working version of setuptools. This must happen before we install
# absl-py, which uses install_requires notation introduced in setuptools 20.5.
pip2 install --upgrade setuptools==39.1.0
pip3 install --upgrade setuptools==39.1.0
pip2 install virtualenv
pip3 install virtualenv
# Install six and future.
pip2 install --upgrade six==1.12.0
pip3 install --upgrade six==1.12.0
pip2 install "future>=0.17.1"
pip3 install "future>=0.17.1"
# Install absl-py.
pip2 install --upgrade absl-py
pip3 install --upgrade absl-py
# Install werkzeug.
pip2 install --upgrade werkzeug==0.11.10
pip3 install --upgrade werkzeug==0.11.10
# Install bleach. html5lib will be picked up as a dependency.
pip2 install --upgrade bleach==2.0.0
pip3 install --upgrade bleach==2.0.0
# Install markdown.
pip2 install --upgrade markdown==2.6.8
pip3 install --upgrade markdown==2.6.8
# Install protobuf.
pip2 install --upgrade protobuf==3.6.1
pip3 install --upgrade protobuf==3.6.1
# Remove obsolete version of six, which can sometimes confuse virtualenv.
@ -71,27 +60,20 @@ rm -rf /usr/lib/python3/dist-packages/six*
# https://github.com/tensorflow/tensorflow/issues/6968
# This workaround isn't needed for Ubuntu 16.04 or later.
if $(cat /etc/*-release | grep -q 14.04); then
pip2 install --no-binary=:all: --upgrade numpy==1.14.5
pip3 install --no-binary=:all: --upgrade numpy==1.14.5
else
pip2 install --upgrade numpy==1.14.5
pip3 install --upgrade numpy==1.14.5
fi
pip2 install scipy==1.2.2
pip3 install scipy==1.4.1
pip2 install scikit-learn==0.18.1
pip3 install scikit-learn==0.18.1
# pandas required by `inflow`
pip2 install pandas==0.19.2
pip3 install pandas==0.19.2
# Benchmark tests require the following:
pip2 install psutil
pip3 install psutil
pip2 install py-cpuinfo
pip3 install py-cpuinfo
# pylint==1.6.4 requires python-astroid (>= 1.4.5) requires lazy-object-proxy
@ -99,57 +81,40 @@ pip3 install py-cpuinfo
# when using setuptools 39.1.0.
# NOTE: Using the updated version of pylint for python3 as python2 is EOL,
# thus using the updated version of lazy-object-proxy==1.4.3
pip2 install lazy-object-proxy==1.4.1
pip3 install lazy-object-proxy==1.4.3
# pylint tests require the following version. pylint==1.6.4 hangs erratically,
# thus using the updated version of 2.5.3 only for python3 as python2 is EOL
# and this version is not available.
pip2 install pylint==1.6.4
pip3 install pylint==2.5.3
# pycodestyle tests require the following:
pip2 install pycodestyle
pip3 install pycodestyle
# tf.mock require the following for python2:
pip2 install mock
pip2 install portpicker
pip3 install portpicker
# TensorFlow Serving integration tests require the following:
pip2 install grpcio
pip3 install grpcio
# Eager-to-graph execution needs astor, gast and termcolor:
pip2 install --upgrade astor
pip3 install --upgrade astor
pip2 install --upgrade gast
pip3 install --upgrade gast
pip2 install --upgrade termcolor
pip3 install --upgrade termcolor
# Keras
pip2 install keras_preprocessing==1.1.0 --no-deps
pip3 install keras_preprocessing==1.1.0 --no-deps
pip2 install --upgrade h5py==2.8.0
pip3 install --upgrade h5py==3.1.0
# Estimator
pip2 install tf-estimator-nightly --no-deps
pip3 install tf-estimator-nightly --no-deps
# Tensorboard
pip2 install tb-nightly --no-deps
pip3 install tb-nightly --no-deps
# Argparse
pip2 install --upgrade argparse
pip3 install --upgrade argparse
# tree
pip2 install dm-tree
pip3 install dm-tree
# tf.distribute multi worker tests require the following: