Как убить все процессы, использующие данный графический процессор?

возможно, это то, что вам нужно:

kill -9 $(nvidia-smi | awk '$2 == "GPU" && $3 == "PID" {flag = 1} flag && $3 > 0 {print $2, $3}' | awk '$1 == 2 {print $2}')

Для более сложных условий вы можете изменить условия второй команды awk. Например, следующую команду можно использовать для уничтожения всех процессов, использующих GPU -0 до GPU -3 и PID > 1000:

kill -9 $(nvidia-smi | awk '$2 == "GPU" && $3 == "PID" {flag = 1} flag && $3 > 0 {print $2, $3}' | awk '$1 < 4 && $2 > 1000 {print $2}')

Как видите, kill -9 PIDsнужны PID для завершения процессов, а awkиспользуется дважды, чтобы найти допустимые PID для уничтожения.

В частности, первая команда awkвыберет строки после строки «GPU PID Type Process Name», а затем распечатает строки идентификаторов GPU и PID с пробелом между каждым идентификатором GPU и PID. Второй awkнайдет определенные идентификаторы GPU или PID, что в данном случае означает поиск всех процессов, использующих GPU -2, а затем распечатку PID. Наконец, kill -9 PIDsуничтожит процессы в соответствии с этими PID.

lsof /dev/nvidia* | awk '{print $2}' | xargs -I {} kill {}

сработало для меня.

В моем случае процессы не были перечислены из nvidia-smi.

Ссылка:https://stackoverflow.com/questions/4354257/can-i-stop-all-processes-using-cuda-in-linux-without-rebooting