原因是在gpu下超出显存,不得已在cpu下训练模型
该链接中有,
点击打开链接
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
要在import keras之前加上上述代码
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
要在import keras之前加上上述代码