How to use tensorflow multinode via slurm?