【发布时间】: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