1.我已经在GPU上训练了我的深度学习模型，使用的是
import keras model.save('trained_model.h5')-
我想在CPU上加载模型并进行预测，使用的是
model = load_model(‘trained_model.h5’)
但是遇到了以下错误
tensorflow.python.framework.errors_impl.InternalError: CUDA runtime implicit initialization on GPU:0 failed. Status: out of memory如何在CPU上加载训练好的模型？
回答：
您收到的错误代码表示您的GPU内存不足。要在CPU上运行，您需要在终端中输入
CUDA_VISIBLE_DEVICES=""或者在代码中使用
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'