tensorflow、pytorch在Jupyter Notebook上自動釋放GPU資源的方法

在使用TensorFlow或pytorch的時候，Jupyter Notebook在運行結束後不會釋放GPU資源，就算把session關閉了也不會釋放。這時候我們就需要手動關停kernel來釋放資源。當然有種在notebook內用命令行釋放資源的方法。代碼如下

import os

pid = os.getpid()
!kill -9 $pid

在感嘆號！後面加上命令行指令就可以在notebook內部執行shell命令。