特定のGPUを使用してすべてのプロセスを終了するには?

特定のGPUを使用してすべてのプロセスを終了するには?

私はNvidia GPUでいくつかの計算を実行するためにCUDAツールキットを使用しています。特定のGPUを使用してすべてのプロセスを終了するには? (すぐに終了します。つまり、後続のPIDを手動で入力する必要はありませんkill -9。)

たとえば、GPU 2を使用してすべてのプロセスを終了します。

ここに画像の説明を入力してください。

ベストアンサー1

Unixの哲学に応じて、特定のGPUを使用してプロセスを一覧表示するツールと、プロセスを終了するツールがあります。シェル構成とテキスト処理ツールを使用して組み合わせます。

たとえば、GPU 2 を使用してすべてのプロセスを終了するには、次のコマンドを実行できます。

kill $(nvidia-smi | awk '$2=="Processes:" {p=1} p && $2 == 2 && $3 > 0 {print $3}')

または

kill $(nvidia-smi -g 2 | awk '$2=="Processes:" {p=1} p && $3 > 0 {print $3}')

おすすめ記事