diff options
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} |