tensorflow_on_slurm

Slurm上的Tensorflow 该软件包使在Slurm集群上运行分布式TensorFlow作业变得更加容易。 它包含用于解析Slurm环境变量以便为分布式TF创建配置的函数。 先决条件 您需要安装TensorFlow。 所有示例均已使用TensorFlow 1.0.1进行了测试，但其他版本也有很好的机会正常工作。 安装 要安装，请在命令行上执行以下命令： git clone cd tensorflow_on_slurm sudo pip install . 用法 示例目录中包含使用CIFAR-10数据集的完整用法示例。 但是，如果您只是想潜入，则可以将以下代码段粘贴到脚本中： import tensorflow as tf from tensorflow_on_slurm import tf_config_from_slurm cluster , my_job_name , m