学习环境:3060+cuda11.3+cudnn8.2.0+torch1.9+tensorflow2.4.0
1、问题原因:
tensorflow的版本有问题
tensorflow1系列不用tensorflow2系列来跑代码
2、解决办法:
安装tensorflow1.15版本:
(1)安装 TensorFlow wheel的索引
pip install nvidia-pyindex
(2)安装TensorFlow1.15
pip install nvidia-tensorflow
(3)检查是否安装成功
python
import tensorflow as tf
tf .__version__
print('Num GPUs Available:', len(tf.config.experimental.list_physical_devices('GPU')))
3、完结撒花