2020年5月7日
win10+keras 跑ResNet 完美解决Failed to get convolution algorithm. This is probably because cuDNN failed to initialize报错
摘要： #GPU按需分配,解决 import tensorflow as tf import keras config = tf.ConfigProto() config.gpu_options.allow_growth = True keras.backend.tensorflow_backend.set 阅读全文
posted @ 2020-05-07 18:11 月夜_1 阅读(1227) 评论(0) 推荐(0) 编辑