cluster = tf.train.ClusterSpec({'your_taskname': tf_hostlist})
server = tf.train.Server(cluster.as_cluster_def(), job_name='your_taskname',
    task_index=task_index)
Запуск TensorFlow в кластере Slurm?