【踩坑】Jupyter调用TensorFlow 导致kernel restarting

303 阅读1分钟

原因一: GPU内存满了。

解决方法: 手动清理一下,kill掉jupyter进程。

原因二: 装了GPU版的TensorFlow,但自己的电脑是CPU的。

解决方法: 重装一下。

pip uninstall tensorflow
pip install tensorflow-cpu

原因三: 缺少个包。

解决方法: 安装一下。

conda install nomkl

原因四: CUDA版本太高了

解决方法: 代码前面添加环境变量。

import os
os.environ['KMP_DUPLICATE_LIB_OK']='True'

以上建议依次试一下。

如果都不行,那可能就是keras和TensorFlow的版本兼容问题

 

pip install keras==2.6.0
pip install tensorflow==2.6.0