我正在训练一个CNN模型来分类简单的图像(正方形和十字形),当我使用CPU时一切都很好,但当我使用GPU时,一切都很好,直到训练开始,我得到这个错误:2022-06-15 04:25:49. 158944:已加载的cuDNN版本8401然后程序就停了。有人知道怎么解决这个问题吗?
kwvwclae1#
如果您使用pycharm,您可以选择“在输出控制台中仿真终端”选项来打印详细的错误信息。运行-〉编辑配置-〉执行-〉在输出控制台中仿真终端在Windows上,也许CUDA缺少zlibwapi.dll文件,您可以下载它并将其移动到CUDA的bin中。
1条答案
按热度按时间kwvwclae1#
如果您使用pycharm,您可以选择“在输出控制台中仿真终端”选项来打印详细的错误信息。
运行-〉编辑配置-〉执行-〉在输出控制台中仿真终端
在Windows上,也许CUDA缺少zlibwapi.dll文件,您可以下载它并将其移动到CUDA的bin中。