tensorflow、pytorch在Jupyter Notebook上自动释放GPU资源的方法
jupyter notebook上释放GPU资源
·
在使用TensorFlow或pytorch的时候，Jupyter Notebook在运行结束后不会释放GPU资源，就算把session关闭了也不会释放。这时候我们就需要手动关停kernel来释放资源。当然有种在notebook内用命令行释放资源的方法。代码如下
import os
pid = os.getpid()
!kill -9 $pid在感叹号！后面加上命令行指令就可以在notebook内部执行shell命令。
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)