web-dev-qa-db-ja.com

キーワードを使用してnvidia-smiでPIDを持つGPUのプロセスを強制終了する方法は?

ターミナルで特定のプログラム(Pythonなど)のGPUで実行中のプロセスを強制終了するにはどうすればよいですか?たとえば、2つのプロセスは、上の画像でpythonで実行されており、nvidia-smiで下の画像を見るためにそれらを強制終了します

For example two processes are running with python in the top picture and kill them to see the bottom picture in nvidia-smi

8
salehinejad

Nvidia-smiでpythonをgrapしてから、次のようなPIDをkill -9コマンドに渡すことができます。

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

5
salehinejad

おそらくnvidia-smiが異なるバージョン/ハードウェア間で異なる形式を持っているため、受け入れられた答えは私にとってはうまくいきません。

私はずっときれいなコマンドを使用しています:

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

Awk式の$3を置き換えて、nvidia-smi出力に合わせることができます。 PIDが発生するのはn番目の列です。

7
Ainz Titor