keras failed to create cublas handle:CUBLAS_STATUS_ALLOC_FAILED问题
Keras深度学习时报错 failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
需要添加一下代码到文件中:
from keras.backend.tensorflow_backend import set_session
import tensorflow as tf
config = tf.ConfigProto()
config....
原创
2019-10-28 22:20:42 ·
543 阅读 ·
0 评论