(https://www.tensorflow.org/api_docs/python/tf/data/experimental/service)
      to write distributed dataset snapshots. The call is non-blocking and
      returns without waiting for the snapshot to finish. Setting `wait=True` to
      `tf.data.Dataset.load` allows the snapshots to be read while they are
      being written.

### Bug Fixes and Other Changes

* <SIMILAR TO ABOVE SECTION, BUT FOR OTHER IMPORTANT CHANGES / BUG FIXES>

  proc = subprocess.Popen(
      [environ_cp['PYTHON_BIN_PATH'], paths[0]] + cuda_libraries,
      stdout=subprocess.PIPE,
      env=maybe_encode_env(environ_cp))

  if proc.wait():
    # Errors from find_cuda_config.py were sent to stderr.
    print('Asking for detailed CUDA configuration...\n')
    return False

  config = dict(