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

在使用TensorFlow或pytorch的時候，Jupyter Notebook在運行結束後不會釋放GPU資源，就算把session關閉了也不會釋放。這時候咱們就須要手動關停kernel來釋放資源。固然有種在notebook內用命令行釋放資源的方法。代碼以下python import os pid = os.getpid() !kill -9 $pid 在感嘆號！後面加上命令行指令就能夠在no

>>阅读原文<<