PyTorchでGPUメモリが解放されないときの対処法

PyTorchのDataLoaderのバグでGPUメモリが解放されないことがある．
nvidia-smiで見ても該当プロセスidは表示されない．
下のコマンドで無理やり解放できる．

ps aux|grep <username>|grep python|awk '{print $2}'|xargs kill