时间戳

首页 > 解决方案 > 如何使用 TensorFlow GPU？

python - 如何使用 TensorFlow GPU？

问题描述

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

import tensorflow as tf

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

您的 CPU 支持未编译此 TensorFlow 二进制文件以使用的指令：AVX2

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

如何切换到 GPU 版本？

标签： pythontensorflowgpu

解决方案

按照本教程Tensorflow GPU我做到了，它运行良好。

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

脚步：

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

推荐阅读