训练yolo3_keras遇到Keras Tensor flow CUBLAS_STATUS_NOT_INITIALIZED解决方法

最新推荐文章于 2024-07-01 15:52:43 发布

zhuoyueljl

最新推荐文章于 2024-07-01 15:52:43 发布

阅读量341

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/zhuoyueljl/article/details/89337550

版权

深度学习专栏收录该内容

10 篇文章 1 订阅

订阅专栏

试过

sudo rm -rf ~/.nv/

，不能解决问题

将下面的程序加到你的代码中即可，问题解决

import tensorflow as tf  
from keras.backend.tensorflow_backend import set_session  
config = tf.ConfigProto()  
config.gpu_options.allow_growth = True  # dynamically grow the memory used on the GPU  
config.log_device_placement = True  # to log device placement (on which device the operation ran)  
                                    # (nothing gets printed in Jupyter, only if you run it standalone)
sess = tf.Session(config=config)  
set_session(sess)  # set this TensorFlow session as the default session for Keras