这个问题也是遇到第二次了,才来记下来;
同时用pip安装了tensorflow和tensorflow-gpu,可能会默认使用tensorflow;
然后需要 pip uninstall keras tensorflow tensorflow-gpu
然后再重新安装 pip install --upgrade tensorflow-gpu keras
--upgrade 很重要
安装的时候可以选择国内镜像加速 后加 -i https://pypi.tuna.tsinghua.edu.cn/simple