message: Set TF_NEED_ROCM=0 when building with CUDA.
diff: @@ -55,6 +55,7 @@ function build_libtensorflow_tarball() {
  export CC_OPT_FLAGS='-mavx'
  if [ "${TF_NEED_CUDA}" == "1" ]; then
  BAZEL_OPTS="${BAZEL_OPTS} --config=cuda"
  + export TF_NEED_ROCM=0
  fi
  bazel clean --expunge
  yes "" | ./configure
  