如何在keras中使用TensorFlow后端而不更改keras.json?

时间:2017-11-03 21:07:44

标签: python tensorflow keras ubuntu-16.04

好吧,我正在研究GPU服务器。所有其他用户正在使用keras Theano backend

我使用TensorFlow安装了virtualenv,之后我使用通常的pip命令安装了keras。当我尝试在此虚拟环境中运行keras时,由于Theano文件,它正在使用/home/user/.keras/keras.json后端。我知道我可以通过在json文件中进行更改来使用TensorFlow,但由于某些原因,其他用户不允许这样做。

我想知道是否有办法在我的虚拟环境中安装keras TensorFlow而不是全局安装的"Theano"

注意:其他用户仍应该可以使用Theano

操作系统:Ubuntu 16.04

2 个答案:

答案 0 :(得分:2)

使用@GPhilo文档参考;

我在代码的开头添加了以下行,成功使用了TensorFlow后端:

import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
# rest of the code

确保您已激活虚拟环境。

答案 1 :(得分:0)

来自Keras' documentation

  

您还可以定义环境变量KERAS_BACKEND和此   将覆盖配置文件中定义的内容:

     

KERAS_BACKEND=tensorflow python -c "from keras import backend" Using TensorFlow backend.