会员
周边
新闻
博问
AI培训
云市场
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
BooTurbo
博客园
首页
新随笔
联系
管理
订阅
2020年8月9日
神经网络训练时出现 无法获取卷积算法的问题 tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
摘要: 使用训练环境: Tensorflow 2.2.0,CUDA 10.1,cuDNN 7.6.5。 当实例化VGG16卷积基时,出现了无法获取卷积算法的问题,提示可能因为cuDNN无法初始化。 打开代码文件在前面添加如下行,问题得以解决。 from tensorflow.compat.v1 import
阅读全文
posted @ 2020-08-09 16:47 BooTurbo
阅读(3352)
评论(1)
推荐(1)
编辑
公告