kerasGPU训练模型问题:Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
最新推荐文章于 2021-08-25 10:46:38 发布
当在训练模型时遇到'Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR'的问题,本文提供了一种针对keras用户的解决方案。不需要在创建会话时添加代码,而是通过修改tensorflow的session.py源码来解决。具体步骤包括删除~/.nv/文件夹,以及在Session类的初始化函数中插入特定代码。完成修改后,错误将被消除。
摘要由CSDN通过智能技术生成