Could not create cudnn handle: CUDNN_STATUS_INTERNAL

本文链接：https://blog.csdn.net/a694262054/article/details/108373103

环境：ubuntu 20.04

tensorflow-gpu=1.15

cuda=10.10

cudnn=7.6

显卡：RTX2070

主要问题：

2020-09-02 23:35:17.561688: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 7215 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2070, pci bus id: 0000:01:00.0, compute capability: 7.5)
2020-09-02 23:35:17.635400: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libcudnn.so.7
2020-09-02 23:35:18.335566: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2020-09-02 23:35:18.344331: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
Traceback (most recent call last):
File "/home/lbz/anaconda3/envs/Compress/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1365, in _do_call
    return fn(*args)
File "/home/lbz/anaconda3/envs/Compress/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1350, in _run_fn
    target_list, run_metadata)
File "/home/lbz/anaconda3/envs/Compress/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
(0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
   [[{{node analysis/layer_0/convolution}}]]
(1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
   [[{{node analysis/layer_0/convolution}}]]
   [[entropy_model/entropy_model_2/compress/strided_slice/_15]]
0 successful operations.
0 derived errors ignored.

解决方法：

在程序中加入：

gpus= tf.config.experimental.list_physical_devices('GPU')
# gpus = tf.config.list_physical_devices('GPU')  # tf2.1版本该函数不再是experimental
print(gpus)  # 前面限定了只使用GPU1(索引是从0开始的,本机有2张RTX2080显卡)
tf.config.experimental.set_memory_growth(gpus[0], True)  # 其实gpus本身就只有一个元素

来源:https://blog.csdn.net/ouening/article/details/104011542

https://github.com/tensorflow/tensorflow/issues/24496#