我正在用keras训练我的模型.当我比较GPU与CPU的性能时.CPU版本比GPU版本快得多
我如何解决以下这些错误?
我试图强制使用tensorflow到GPU,我得到这些错误:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device to node 'gradients/simple_rnn_1/while/Select_1_grad/Select/f_acc':
Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
Colocation Debug Info:
Colocation group had the following types and devices:
Tile: CPU
StackPush: GPU CPU
Relu: GPU CPU
ReluGrad: GPU CPU
ZerosLike: GPU CPU
Select: GPU CPU
StackPop: GPU CPU
AddN: GPU CPU
RefEnter: GPU CPU
Stack: GPU CPU
Run Code Online (Sandbox Code Playgroud)
我的模型看起来像这样:
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
masking_1 (Masking) (None, None, …Run Code Online (Sandbox Code Playgroud)