首先，您需要安装tensorflow-gpu ，因为此软件包负责gpu计算。还记得运行环境变量CUDA_VISIBLE_DEVICES = 0来运行代码（或者，如果您有多个GPU，则用逗号将它们的索引放入）。可能与使用gpu有关。如果您的张量流仍然不使用gpu，请尝试此操作

遵循本教程Tensorflow GPU，我做到了，它运行完美。

注意！ -安装9.0版！ Tensorflow-gpu不支持较新的版本

脚步：

1. 卸载旧的tensorflow
2. 安装tensorflow-gpu pip install tensorflow-gpu
3. 安装Nvidia显卡和驱动程序（您可能已经拥有）
4. 下载并安装CUDA
5. 下载并安装cuDNN
6. 通过简单程序验证

from tensorflow.python.client import device_lib print(device_lib.list_local_devices())