具有tensorflow的keras：可能在不使用StreamExecutor上下文的情况下执行了CUDA运行时调用

Question

我正在使用具有张量流后端的kears，以下是问题所在。有没有可以解决这个问题，谢谢！

Answer 1

错误是由CNMEM的非法值引起的。根据theano doc，CNMEM只能被指定为浮点数。

  

0：未启用。

     

0＆lt; N <= 1：使用总GPU内存的这一部分（为驱动程序内存剪裁为.95）。

     

1：以兆字节（MB）的内存使用此数字。

您也可以参考here。

警告是由于Theano（Kera的后端）的变化。它将从CUDA变为GpuArray。您可以参考here获取解决方案。

实际上，如果您修正了警告，错误也将根据以下内容消失：

  

此值仅在使用（CUDA后端）时分配GPU内存，在GPU后端为（GpuArray后端）时无效。对于新的后端，请参阅config.gpuarray.preallocate