Pytorch kill processes distributed on multiple Gpus

kill $(ps aux | grep “train.py” | grep -v grep | awk ‘{print $2}’)

From frankang’s answer

https://github.com/pytorch/fairseq/issues/487