Blas GEMM launch failed 错误解决方案

错误信息如下：

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)