使用keras训练CNN模型可以使用CPU,但不能使用GPU

wz3gfoph  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(319)

我正在训练一个CNN模型来分类简单的图像(正方形和十字形),当我使用CPU时一切都很好,但当我使用GPU时,一切都很好,直到训练开始,我得到这个错误:
2022-06-15 04:25:49. 158944:已加载的cuDNN版本8401
然后程序就停了。有人知道怎么解决这个问题吗?

kwvwclae

kwvwclae1#

如果您使用pycharm,您可以选择“在输出控制台中仿真终端”选项来打印详细的错误信息。
运行-〉编辑配置-〉执行-〉在输出控制台中仿真终端
在Windows上,也许CUDA缺少zlibwapi.dll文件,您可以下载它并将其移动到CUDA的bin中。

相关问题