错误信息如下:
tensorflow.python.framework.errors_impl.InternalError: 2 root error(s) found.
(0) Internal: Blas GEMM launch failed : a.shape=(10, 10), b.shape=(10, 10), m=10, n=10, k=10
[[{{node sequential/simple_rnn/while/body/_1/MatMul_1}}]]
[[Reshape_6/_32]]
(1) Internal: Blas GEMM launch failed : a.shape=(10, 10), b.shape=(10, 10), m=10, n=10, k=10
[[{{node sequential/simple_rnn/while/body/_1/MatMul_1}}]]
解决方案:
physical_devices = tf.config.list_physical_devices('GPU') tf.config.experimental.set_memory_growth(physical_devices[0], True)