跑深度学习模型报错：Function call stack:train_function -＞ train_function

最新推荐文章于 2024-09-11 09:52:07 发布

傅华涛Fu

最新推荐文章于 2024-09-11 09:52:07 发布

阅读量4.4k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：深度学习 Keras TensorFlow 文章标签：深度学习人工智能

本文链接：https://blog.youkuaiyun.com/fu_jian_ping/article/details/121062786

深度学习同时被 3 个专栏收录

42 篇文章

订阅专栏

Keras

10 篇文章

订阅专栏

TensorFlow

10 篇文章

订阅专栏

今天在跑自己设计的深度学习模型时出现了以下问题：

InternalError: 2 root error(s) found.
  (0) Internal:  Blas GEMM launch failed : a.shape=(12, 3), b.shape=(3, 32), m=12, n=32, k=3
	 [[node model/dense_3/MatMul (defined at <ipython-input-16-5a73e5716594>:10) ]]
	 [[gradient_tape/model/token_and_position_embedding/embedding_1/embedding_lookup/Reshape/_98]]
  (1) Internal:  Blas GEMM launch failed : a.shape=(12, 3), b.shape=(3, 32), m=12, n=32, k=3
	 [[node model/dense_3/MatMul (defined at <ipython-input-16-5a73e5716594>:10) ]]
0 successful operations.
0 derived errors ignored. [Op:__inference_train_function_3032]

Function call stack:
train_function -> train_function

出现该问题的原因可能是各种库之间的版本依赖问题，或者是GPU被占用了。

而我的情况是GPU被占用了，因此我把其他占用GPU的程度都关了，再次跑模型，OK了！