没有GPU的Unity ML-Agents的Tensorflow和Cuda

时间:2018-01-17 00:55:46

标签: python tensorflow ml-agent

我正在尝试使用tensorflow与unity ml-agents一起使用。我按照设置here的设置说明进行操作,直至“创建conda环境”。我设法成功运行conda create命令,激活tensorflow,安装tensorflow,最后启动python。

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

任何人都可以告诉我这应该如何出现在环境变量界面中吗?或者有没有人知道如何正确设置它?

提前致谢

1 个答案:

答案 0 :(得分:0)

卸载tensorflow

  

Pip卸载tensorflow

  

pip卸载tensorflow-gpu

取决于您安装的内容 然后

  

pip安装tensorflow

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

  

pip install tensoflow == [版本]