Как убить процесс на графических процессорах с PID в nvidia-smi, используя ключевое слово?

Question

Как убить запущенные процессы на графических процессорах для конкретной программы (например, python) в терминале? Например, два процесса выполняются с python на верхней картинке и убивают их, чтобы увидеть нижнюю картинку в nvidia-smi

Answer 1

Принятый ответ не работает для меня, возможно, потому что nvidia-smi имеет разные форматы в разных версиях / оборудовании.

Я использую намного более чистую команду:

nvidia-smi | grep 'python' | awk '{ print $3 }' | xargs -n1 kill -9

Вы можете заменить $3 в выражении awk, чтобы соответствовать вашему выводу nvidia-smi. Это n-й столбец, в котором встречаются PID.

Answer 2

Вы можете захватить python в nvidia-smi, а затем передать PID команде kill -9, например:

sudo kill -9 $ (nvidia-smi | grep 'python' | sed -n 's / | \ s * [0-9] \ s ([0-9] ) \ s . * / \ 1 / p' | sed '/ ^ $ / d')