Skip to content

Commit 2df5ec1

Browse files
yifeiftensorflower-gardener
authored andcommitted
Fix pip wheel hash mismatch issue for tf-nightly. tensorflow#33569
PiperOrigin-RevId: 278012174 Change-Id: I56b1cdaeb297ac49eecc8074eacd087e67d79de9
1 parent 30ac8e4 commit 2df5ec1

File tree

4 files changed

+8
-20
lines changed

4 files changed

+8
-20
lines changed

tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/nightly_release.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,9 @@ yes "" | "$PYTHON_BIN_PATH" configure.py
4242
bazel build --config=opt --config=v2 \
4343
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \
4444
tensorflow/tools/pip_package:build_pip_package
45-
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
4645

47-
# Copy and rename to tf_nightly
48-
for WHL_PATH in $(ls pip_pkg/tf_nightly_gpu-*dev*.whl); do
49-
copy_to_new_project_name "${WHL_PATH}" tf_nightly
50-
done
46+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --nightly_flag
47+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
5148

5249
# Upload the built packages to pypi.
5350
for WHL_PATH in $(ls pip_pkg/tf_nightly*dev*.whl); do

tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/nightly_release.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,9 @@ yes "" | "$PYTHON_BIN_PATH" configure.py
4242
bazel build --config=opt --config=v2 \
4343
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \
4444
tensorflow/tools/pip_package:build_pip_package
45-
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
4645

47-
# Copy and rename to tf_nightly
48-
for WHL_PATH in $(ls pip_pkg/tf_nightly_gpu-*dev*.whl); do
49-
copy_to_new_project_name "${WHL_PATH}" tf_nightly
50-
done
46+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --nightly_flag
47+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
5148

5249
# Upload the built packages to pypi.
5350
for WHL_PATH in $(ls pip_pkg/tf_nightly*dev*.whl); do

tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/nightly_release.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,9 @@ yes "" | "$PYTHON_BIN_PATH" configure.py
4242
bazel build --config=opt --config=v2 \
4343
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \
4444
tensorflow/tools/pip_package:build_pip_package
45-
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
4645

47-
# Copy and rename to tf_nightly
48-
for WHL_PATH in $(ls pip_pkg/tf_nightly_gpu-*dev*.whl); do
49-
copy_to_new_project_name "${WHL_PATH}" tf_nightly
50-
done
46+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --nightly_flag
47+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
5148

5249
# Upload the built packages to pypi.
5350
for WHL_PATH in $(ls pip_pkg/tf_nightly*dev*.whl); do

tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nightly_release.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,9 @@ yes "" | "$PYTHON_BIN_PATH" configure.py
4242
bazel build --config=opt --config=v2 \
4343
--crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain \
4444
tensorflow/tools/pip_package:build_pip_package
45-
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
4645

47-
# Copy and rename to tf_nightly
48-
for WHL_PATH in $(ls pip_pkg/tf_nightly_gpu-*dev*.whl); do
49-
copy_to_new_project_name "${WHL_PATH}" tf_nightly
50-
done
46+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --nightly_flag
47+
./bazel-bin/tensorflow/tools/pip_package/build_pip_package pip_pkg --gpu --nightly_flag
5148

5249
# Upload the built packages to pypi.
5350
for WHL_PATH in $(ls pip_pkg/tf_nightly*dev*.whl); do

0 commit comments

Comments
 (0)