您的位置:首页 > 大数据 > 人工智能

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

2020-04-24 11:02 1226 查看

keras使用gpu加速训练时出现了以下问题

UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
[[node conv1_9/convolution (defined at E:\anaconda\lib\site-packages\keras\backend\tensorflow_backend.py:3335) ]]
[[node metrics_6/acc/Mean (defined at E:\anaconda\lib\site-packages\keras\backend\tensorflow_backend.py:1367) ]]

搜到的解决方法1:

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

未能解决问题

搜到的方法2:

import tensorflow as tf
import keras
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

解决问题

  • 点赞
  • 收藏
  • 分享
  • 文章举报
lwh12138 发布了2 篇原创文章 · 获赞 0 · 访问量 232 私信 关注
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐