『开发技巧』解决RTX 2060 TensorFlow CUDNN_STATUS_INTERNAL_ERROR错误

最新推荐文章于 2024-06-26 20:53:54 发布

小宋是呢

最新推荐文章于 2024-06-26 20:53:54 发布

阅读量688

点赞数

分类专栏：开发技巧

本文链接：https://blog.csdn.net/xiaosongshine/article/details/107475513

版权

开发技巧专栏收录该内容

38 篇文章 1 订阅

订阅专栏

问题描述

在使用RTX 2060 开发TensorFlow 1x 版本遇到一个错误，如下：

2020-07-20 22:08:55.555961: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] 
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2020-07-20 22:08:55.571109: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] 
Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

解决方法

解决方法在头部加入如下：

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession


config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)