1. 首页
  2. 数据库
  3. 其它
  4. Failed to get convolution algorithm. This is probably because cuDNN failed to in

Failed to get convolution algorithm. This is probably because cuDNN failed to in

上传者: 2021-01-15 05:35:28上传 PDF文件 22.28KB 热度 15次
使用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_o
用户评论