From 077ae24c53157dd889474baea4eee89acdbf629a Mon Sep 17 00:00:00 2001
From: Yifei Feng <yifeif@google.com>
Date: Fri, 5 Jun 2020 11:49:07 -0700
Subject: [PATCH] Install python3 package by default

PiperOrigin-RevId: 314967616
Change-Id: Ibb43020f86b4eee2df53727774726d44e074c8dd
---
 tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16        | 2 +-
 .../tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.0      | 2 +-
 .../tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.1      | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16 b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16
index c540d03df7f..305c5a7c381 100644
--- a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16
+++ b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16
@@ -75,7 +75,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.6 0
 # Install given tensorflow or tf-nightly version, if not specified, install the # latest official release
 ARG TF_PACKAGE=tensorflow
 ARG TF_PACKAGE_VERSION=
-RUN pip install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
+RUN pip3 install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
 
 # TODO(klimek): Figure out a better way to get the right include paths
 # forwarded when we install new packages.
diff --git a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.0 b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.0
index 1a1200095b1..0732cd834a3 100644
--- a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.0
+++ b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.0
@@ -75,7 +75,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.6 0
 # Install given tensorflow or tf-nightly version, if not specified, install the # latest official release
 ARG TF_PACKAGE=tensorflow
 ARG TF_PACKAGE_VERSION=
-RUN pip install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
+RUN pip3 install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
 
 # TODO(klimek): Figure out a better way to get the right include paths
 # forwarded when we install new packages.
diff --git a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.1 b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.1
index 01f34365ff2..4d58ad67df6 100644
--- a/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.1
+++ b/tensorflow/tools/ci_build/Dockerfile.custom_op_ubuntu_16_cuda10.1
@@ -77,7 +77,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.6 0
 # Install given tensorflow or tf-nightly version, if not specified, install the # latest official release
 ARG TF_PACKAGE=tensorflow
 ARG TF_PACKAGE_VERSION=
-RUN pip install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
+RUN pip3 install ${TF_PACKAGE}${TF_PACKAGE_VERSION:+==${TF_PACKAGE_VERSION}}
 
 # TODO(klimek): Figure out a better way to get the right include paths
 # forwarded when we install new packages.