原因:tensorflow-gpu版本不匹配
法1:更换tensorflow-gpu版本
命令行输入如下代码,查看原tensorflow-gpu版本,得到tensorflow-gpu版本为2.1.0,但所运行代码需要1.12.0版本
python
import tensorflow as tf
tf.__version__
运行如下代码即可解决tensorflow版本问题,==后面接想要的版本号:
pip uninstall tensorlow-gpu
pip install tensorflow-gpu==1.12.0
法2:安装合适的版本的cuda
tensorflow各个版本需要的CUDA版本以及Cudnn的对应关系见网站:
https://www.tensorflow.org/install/source#common_installation_problems
https://www.tensorflow.org/install/source_windows