Kill all the processes on a given GPU

killgpu.sh

# usage: `killgpu 0`  will kill all the processes that are using gpu 0
killgpu (){
    # https://forums.developer.nvidia.com/t/11-gb-of-gpu-ram-used-and-no-process-listed-by-nvidia-smi/44459/14
    fuser -v /dev/nvidia$@ 2>&1 | grep python | grep -o -E " [0-9]+ " | xargs kill -9
}