I break model training and gpu remains with full memory and this helps me. But by carefull its also kills python env kernel
pkill -f python