小编Tob*_*ias的帖子

使用LSTM Networks,Keras在GPU上运行缓慢

我正在用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)

python machine-learning deep-learning keras tensorflow

5
推荐指数
1
解决办法
1518
查看次数