lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update PIP package script to include OSS keras as a PIP dependency.

Browse Source 
Also update keras build script to remove keras-nightly package if it is there. This will ensure keras integration test to only run with the code from local workspace.

PiperOrigin-RevId: 364873257
This commit is contained in:
Scott Zhu 2021-03-24 13:04:30 -07:00 committed by TensorFlower Gardener
parent 00135a7069
commit 2ac31074fe
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
keras/kokoro/github/ubuntu/cpu/build.sh
View File

@ -35,6 +35,9 @@ cd "src/github/keras"
# Keep pip version at 20.1.1 to avoid the slow resolver issue.
pip install -U pip==20.1.1 setuptools
pip install -r requirements.txt
# Uninstall the keras-nightly package so that we will only test the version of
# keras code from local workspace.
pip uninstall -y keras-nightly
# TODO(scottzhu): Using --define=use_fast_cpp_protos=false to suppress the
# protobuf build issue for now. We should have a proper solution for this.

3
keras/kokoro/github/ubuntu/gpu/build.sh
View File

@ -35,6 +35,9 @@ cd "src/github/keras"
# Keep pip version at 20.1.1 to avoid the slow resolver issue.
pip install -U pip==20.1.1 setuptools
pip install -r requirements.txt
# Uninstall the keras-nightly package so that we will only test the version of
# keras code from local workspace.
pip uninstall -y keras-nightly
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export TF_CUDA_COMPUTE_CAPABILITIES=6.0