python - 如何使用 TensorFlow GPU？

Question

如何在 Python 3.6 x64 中使用 TensorFlow GPU 版本而不是 CPU 版本？

import tensorflow as tf

Python 正在使用我的 CPU 进行计算。
我可以注意到它，因为我有一个错误:

Your CPU supports instructions that this TensorFlow binary was notcompiled to use: AVX2

我已经安装了 tensorflow 和 tensorflow-gpu。

如何切换到 GPU 版本？

Answer 1

按照本教程 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())