Get the docker container ID containing a specified PID (useful to identify GPU-intensive processes when running nvidia-smi)

findpid.sh

#/bin/bash

if [ -z "$1" ]
then
    echo "Supply a PID as an argument!"
    exit 1
fi

for i in $(docker ps -q)
do
    OUTPUT=$(docker container top "$i" | grep "$1")
    if [ $(docker container top "$i" | grep "$1" | wc -c) -ne 0 ]
    then
        docker ps -a | grep $i
        exit 0
    fi
done