使用CPU预测Keras theano

时间:2019-02-13 20:08:26

标签: python keras neural-network grid theano

我正在大学集群中工作,我正在尝试将keras模型导入分析器中以获得一些预测。

该集群只有CPU,而我无权访问CUDA或任何GPU来运行我的模型并进行预测。

我目前能够通过加载模型json和权重进行预测,但是执行时会收到以下消息:

cc1plus: fatal error: cuda_runtime.h: No such file or directory
compilation terminated.
ERROR:theano.sandbox.cuda] Failed to compile cuda_ndarray.cu

我看到其他人也有同样的错误,但是该解决方案不适用于我,因为我没有/ user / local / cuda /并且无法更改。

我将设备设置为CPU,希望这是问题所在,但事实并非如此:

theano.config.device='cpu'

我该怎么办?

0 个答案:

没有答案