这个屌丝很懒，什么也没留下！

tensorflow、pytorch在Jupyter Notebook上自动释放GPU资源的方法_notebook gpu 清除

作者：你好赵伟 | 2024-02-22 00:08:11

notebook gpu 清除

在使用TensorFlow或pytorch的时候，Jupyter Notebook在运行结束后不会释放GPU资源，就算把session关闭了也不会释放。这时候我们就需要手动关停kernel来释放资源。当然有种在notebook内用命令行释放资源的方法。代码如下


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

在感叹号！后面加上命令行指令就可以在notebook内部执行shell命令。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/127478