【文件属性】：
文件名称：tensorflow_on_slurm
文件大小：8KB
文件格式：ZIP
更新时间：2021-04-30 22:50:47
Python
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
【文件预览】：
tensorflow_on_slurm-master
----examples()
--------cifar-10()
----setup.py(326B)
----README.md(2KB)
----tensorflow_on_slurm()
--------tensorflow_on_slurm.py(2KB)
--------__init__.py(90B)
--------tests()