Tensorflow+InternalError: Blas GEMM launch failed

2024年4月1日 236次阅读 来源: tensorflow

[参考1:]https://stackoverflow.com/questions/37337728/tensorflow-internalerror-blas-sgemm-launch-failed

问题: 多半是session用毕没有及时close, 导致系统和GPU的很大部分被占用过却没有归还, 当前资源便不够了; 或是多个session争用GPU.

解决:

在每次测试想运行的session前加上:

if 'session' in locals() and session is not None:
    print('Close interactive session')
    session.close()

当然最好还是尽可能使用上下文管理器吧, 不适合用的地方记得手动close().

shutdown kernel or 重启计算机 亦可释放归还.

    原文作者：tensorflow
    原文地址: https://www.cnblogs.com/ZhengPeng7/p/7837322.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞