jupyterlab运行tensorflow出现kernel restarting的问题解决

鲜于光辉
2023-12-01

最近从jupyter notebook 转向 jupyterlab,功能多一些。可是遇到了问题:运行tensorflow时出现了kernel restarting 的问题。语句没有被执行,且kernel进入到重启后的状态。反复的检查表明,这个问题仍然是显存冲突造成的,可我在ipynb的前面已经加上了限制显存使用的代码。
测试表明:**jupyterlab在退出的时候不会释放显存。**这应该是个bug。notebook是会释放的。因此执行notebook时成功,而用jupyterlab时失败。
解决方案:nvidia_smi查看显存占用的进程,手工kill之,然后重启kernel,搞定。

 类似资料: