原因: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

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐