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

小宋是呢

发布于 2022-03-07 14:09:03

4980

发布于 2022-03-07 14:09:03

文章被收录于专栏：深度应用

问题描述

在使用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)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

tensorflow