【问题标题】:Installing tensorflow-gpu on windows在 windows 上安装 tensorflow-gpu
【发布时间】:2019-04-02 15:30:39
【问题描述】:

我是 python 新手,我有 python 3.5、windows 10、AVX 5200。我想在 gpu 上安装 tensorflow。我逐步尝试了不同的博客,但我无法在 gpu 上获取 tensorflow,总是收到以下消息:

    Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX

请任何人帮助我如何逐步获得 tensorflow-gpu?我会感谢他的。

【问题讨论】:

  • 您可能成功安装了 tensorflow-gpu,但您的代码仍然可以使用 tensorflow-cpu,例如,由于某些环境设置。当你pip install tensorflow-gpu时,安装成功了吗?
  • @InonPeled 感谢您的回复。是的,它说安装成功完成,但是当我使用 pycharm 在 python 中训练 CNN 模型时,它给了我这个消息。如何让我的 CNN 使用 tensorflow-gpu 训练和评估数据,它非常慢并且卡在我的 tensorflow 上,你能帮帮我吗?
  • @InonPeled 我遵循了所提到的内容,但我得到了:I C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\platform\cpu_feature_guard .cc:137] 您的 CPU 支持未编译此 TensorFlow 二进制文件以使用的指令:AVX 2019-04-02 20:26:49.174284: I C:\tf_jenkins\workspace\rel-win\M\windows-gpu\PY\ 35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1105] 找到具有属性的设备 0:名称:NVS 5200M 主要:2 次要:1 memoryClockRate(GHz):1.344 pciBusID:0000:01:00.0 totalMemory:1.00GiB freeMemory : 823.00MiB

标签: python tensorflow gpu


【解决方案1】:
  1. 您已安装并运行 TF。
  2. 您看到的是警告消息,根本不是错误。你安装了一个预编译包,没有启用AVX指令集,AVX是CPU的SIMD指令集。
  3. 您可以通过运行确保 tensorflow gpu 正常工作:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.close()
  1. 强烈建议使用 ananconda 并使用 conda 安装所有内容。
  2. 找一本书或只是阅读官方指南。

【讨论】:

    猜你喜欢
    • 2018-09-11
    • 2021-08-29
    • 2020-12-20
    • 2018-04-10
    • 2019-06-04
    • 2019-01-03
    • 2018-03-25
    • 1970-01-01
    • 2017-09-29
    相关资源
    最近更新 更多