diff options
author | 2018-08-22 13:19:01 -0700 | |
---|---|---|
committer | 2018-08-22 13:25:57 -0700 | |
commit | 6448db09a76e2803c1a46a7514c4bf82dd7d6261 (patch) | |
tree | fa26aa5464f4114bb05b80e8fbf004164a040b73 /tensorflow/tools/ci_build | |
parent | 2d83b5b5953041cf72f88a7a1abb82a820932277 (diff) |
Windows: Fix pip package installation in GPU build
The GPU pip package name has been renamed to tensorflow_gpu-*.whl
PiperOrigin-RevId: 209817853
Diffstat (limited to 'tensorflow/tools/ci_build')
-rw-r--r-- | tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh index 7ac07872e9..f31b0a64e0 100644 --- a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh +++ b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh @@ -120,7 +120,7 @@ if [[ "$TF_NIGHTLY" == 1 ]]; then fi # Running python tests on Windows needs pip package installed -PIP_NAME=$(ls ${PY_TEST_DIR}/tensorflow-*.whl) +PIP_NAME=$(ls ${PY_TEST_DIR}/tensorflow_gpu-*.whl) reinstall_tensorflow_pip ${PIP_NAME} TF_GPU_COUNT=${TF_GPU_COUNT:-8} |