好吧,我正在研究GPU服务器。所有其他用户正在使用keras
Theano backend
。
我使用TensorFlow
安装了virtualenv
,之后我使用通常的pip
命令安装了keras。当我尝试在此虚拟环境中运行keras
时,由于Theano
文件,它正在使用/home/user/.keras/keras.json
后端。我知道我可以通过在json文件中进行更改来使用TensorFlow
,但由于某些原因,其他用户不允许这样做。
我想知道是否有办法在我的虚拟环境中安装keras
TensorFlow
而不是全局安装的"Theano"
。
注意:其他用户仍应该可以使用Theano
操作系统:Ubuntu 16.04
答案 0 :(得分:2)
使用@GPhilo文档参考;
我在代码的开头添加了以下行,成功使用了TensorFlow
后端:
import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
# rest of the code
确保您已激活虚拟环境。
答案 1 :(得分:0)
您还可以定义环境变量
KERAS_BACKEND
和此 将覆盖配置文件中定义的内容:
KERAS_BACKEND=tensorflow python -c "from keras import backend" Using TensorFlow backend.