How to kill process on GPUs with PID in nvidia-smi using keyword?

python gpu nvidia keyword pid
salehinejad picture salehinejad · May 5, 2018 · Viewed 26.3k times · Source

How to kill running processes on GPUs for a specific program (e.g. python) in terminal? For example two processes are running with python in the top picture and kill them to see the bottom picture in 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

Answer

Ainz Titor picture Ainz Titor · May 4, 2019

The accepted answer doesn't work for me, probably because nvidia-smi has different formats across different versions/hardware.

I'm using a much cleaner command:

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

You can replace $3 in the awk expression to fit your nvidia-smi output. It is the n-th column in which the PIDs occur.

Related questions

How to check if pytorch is using the GPU?

I would like to know if pytorch is using my GPU. It's possible to detect with nvidia-smi if there is any activity from the GPU during the process, but I want something written in a python script. Is there a …

Read More
where is the ./configure of TensorFlow and how to enable the GPU support?

When installing TensorFlow on my Ubuntu, I would like to use GPU with CUDA. But I am stopped at this step in the Official Tutorial : Where exactly is this ./configure ? Or where is my root of source tree. My TensorFlow …

Read More
How to tell if tensorflow is using gpu acceleration from inside python shell?

I have installed tensorflow in my ubuntu 16.04 using the second answer here with ubuntu's builtin apt cuda installation. Now my question is how can I test if tensorflow is really using gpu? I have a gtx 960m gpu. When I …

Read More