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

2022-03-07 14:09:03 浏览数 (1)

问题描述

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

代码语言：javascript复制

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

解决方法

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

代码语言：javascript复制

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


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

tensorflow 开发

0 人点赞