View source on GitHub |
Starts a standard TensorFlow server.
tf.contrib.distribute.run_standard_tensorflow_server(
session_config=None
)
This method parses configurations from "TF_CONFIG" environment variable and starts a TensorFlow server. The "TF_CONFIG" is typically a json string and must have information of the cluster and the role of the server in the cluster. One example is:
TF_CONFIG='{ "cluster": { "worker": ["host1:2222", "host2:2222", "host3:2222"], "ps": ["host4:2222", "host5:2222"] }, "task": {"type": "worker", "index": 1} }'
This "TF_CONFIG" specifies there are 3 workers and 2 ps tasks in the cluster and the current role is worker 1.
Valid task types are "chief", "worker", "ps" and "evaluator" and you can have at most one "chief" and at most one "evaluator".
An optional key-value can be specified is "rpc_layer". The default value is "grpc".
Args | |
---|---|
session_config
|
an optional tf.compat.v1.ConfigProto object. Users can
pass in the session config object to configure server-local devices.
|
Returns | |
---|---|
a tf.distribute.Server object which has already been started.
|
Raises | |
---|---|
ValueError
|
if the "TF_CONFIG" environment is not complete. |