如何使用TensorFlow GPU？
python
tensorflow
24
0

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

import tensorflow as tf

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

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

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

如何切换到GPU版本？

参考资料：
Stack Overflow
收藏
评论
共 2 个回答
高赞 时间 活跃

首先，您需要安装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())

收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题