擅长:python、mysql、java
<p>你的MWE对我来说很好(如果我将<code>, input_shape=(38, 21, 1)</code>添加到第一个卷积层):</p>
<pre><code>import numpy as np
import keras
model = keras.models.Sequential() #Sequential model type.
model.add(keras.layers.Conv2D(filters=1, kernel_size=(3,3), strides = 1, activation="sigmoid", input_shape=(38, 21, 1))) #Convolutional layer.
model.add(keras.layers.Flatten()) #Flatten layer.
model.add(keras.layers.Dense(4)) #Dense layer of 4 units.
model.compile(loss='mean_squared_error', optimizer='adam') #compile model.
y = np.random.rand(2, 4) #Random expected output
x = np.random.rand(2, 38, 21, 1) # Random input.
model.fit(x, y)
</code></pre>
<p>这意味着您的问题必须来自您的系统或安装。在</p>
<p>查看<a href="https://www.tensorflow.org/install/source#tested_build_configurations" rel="nofollow noreferrer">compatibility chart of tensorflow</a>可以看出python、tensorflow和CUDA版本应该是兼容的。在</p>
<p>对于您的配置,建议使用cuDNN版本<code>7.0.x</code>。
您使用的cuDNN版本<code>7.2</code>可能不兼容。
尝试安装/使用cudn<code>7.0.x</code>。在</p>