RTX3080 安装tensorflow-gpu and cudnn

今天要测试一个tensorflow的代码 于是就尝试安装了一下在liunx服务器上

1.我的原始环境是torch1.8.0+cu111 ，初始安装的是tensorflow-gpu==2.4.0

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())

结果输出为False 百度之后发现是cudn和cudnn的版本太低
因为我的conda是换过源的 ，所以输入

conda search xxx # 在当前所有通道查找可以安装的xxx

找到你想用的版本并安装

conda install cudnn/cudatoolkit

测试输出还是False
安装 2.5.0版本的gpu 输出结果

2. 在运行代码的时候又出现 AttributeError: module ‘keras.utils.generic_utils’ has no attribute 'populate_dict_with_module_objec的问题

卸载之后的keras==2.3.0 重新安装
代码可以进行测试 虽然效果不是很好
最后说一句 怪不得现在大部分都用pytorch 就是版本和环境的对应问题就要弄好久