如何使用TensorFlow GPU?
python
tensorflow
17
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())

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

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号