【问题标题】:Tensorflow and Cuda for Unity ML-Agents WITHOUT a GPU没有 GPU 的 Unity ML-Agents 的 Tensorflow 和 Cuda
【发布时间】:2018-01-17 00:55:45
【问题描述】:

我正在尝试使用 tensorflow 与统一 ml-agents 一起使用。我遵循了可以在here 找到的设置说明,就“创建 conda 环境”而言。我成功运行 conda create 命令,激活 tensorflow,安装 tensorflow,最后启动 python。

但是当我运行这个命令“import tensorflow as tf”时,由于我的机器没有 GPU,我得到一个错误。我正在假设使用 CUDA_VISIBLE_DEVICES 环境变量将解决此问题并强制使用 CPU,但我不确定如何输入变量。

谁能告诉我这应该如何出现在环境变量界面中?或者是否有人对如何正确设置有任何其他想法?

提前致谢

【问题讨论】:

  • 如果你没有 GPU,那么使用不依赖 GPU 的 tensorflow 构建。在这里尝试使用CUDA_VISIBLE_DEVICES 将无济于事。
  • 非常感谢您的帮助。您能否推荐一个无需 GPU 且可与 Unity ML-Agents 一起使用的 tensorflow 构建?
  • 好的,我为遇到同样问题的任何人解决了这个问题。只需 pip 安装“tensorflow”而不是“tensorflow-gpu”。

标签: python tensorflow ml-agent


【解决方案1】:

卸载张量流

pip 卸载张量流

pip 卸载 tensorflow-gpu

取决于您安装的内容 那么

点安装张量流

或者如果您想安装特定版本

pip install tensoflow==[版本]

【讨论】:

    猜你喜欢
    • 2020-05-05
    • 1970-01-01
    • 1970-01-01
    • 2018-08-29
    • 2022-07-12
    • 1970-01-01
    • 1970-01-01
    • 2011-04-14
    • 2023-03-14
    相关资源
    最近更新 更多