Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try...

Marmara ·
更新时间:2024-09-20
· 906 次阅读

使用inception-V3批处理文件retrain.bat进行预训练时,发现报错“Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try…”于是打开retrain.py文件,在前面加上如下代码:
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
在cd到retrain.bat文件,成功运行。
参考文章:https://blog.csdn.net/qq_41868689/article/details/98503069


作者:weixin_42076509



this is get this TO cudnn try

需要 登录 后方可回复, 如果你还没有账号请 注册新账号
相关文章