长时间卡在 successfully opened dynamic library libcudnn .so.7

长时间卡在 successfully opened dynamic library libcudnn .so.7

问题描述

在使用tensorflow进行训练或者运行tf.test.is_gpu_available()是的时候，界面长时间卡在 successfully opened dynamic library libcudnn .so.7 successfully opened dynamic library libcudnn .so.10.1 不再输出任何log信息。

相似的问题可见：https://github.com/tensorflow/tensor2tensor/issues/1643

环境

OS: Ubuntu 18.04.2 LTS
CUDA Version 11.4.48
CuDNN Version 8.2.1
Python 3.7
Tensorflow-gpu == 2.4

解决思路

从以上github issues的讨论中，发现大概率是CUDA、cuDNN、tensorflow的版本匹配问题。
在 https://tensorflow.google.cn/install/gpu 上发现只有tensorflow>=2.5.0的版本才支持CUDA11.2，而我安装的是CUDA11.4

解决方案

将原有的tenforflow卸载，下载tensorflow2.5.0

pip install tensorflow-gpu==2.5

这里我的conda install安装失败了，所以用了pip install