【记录】在 Slurm 平台的GPU集群上使用 Pytorch