我用GPU在机器上建立了带有TensorFlow后端的模型。现在我想在没有GPU的生产机器上使用这个型号。在没有安装TensorFlow后端的情况下,我不能使用这个模型,但是我也不能在没有GPU的机器上安装TensorFlow后端。
在没有GPU支持的情况下,可以为TensorFlow构建H2O后端吗?
发布于 2017-05-23 14:54:51
但是,如果没有GPU,我也不能在机器上安装TensorFlow后端。
你是说你试着犯错误了吗?
在没有GPU支持的情况下,可以为TensorFlow构建H2O后端吗?
是的,但取决于你的机器,它可能很容易,也可能有点烦琐。如果您在一个MacOS系统上运行,那么它应该是开箱即用的,只有听从我们的指示。
如果您是在一个基于Linux的系统上,那么您必须首先修改我们的脚本,确切地说,是这一个。你必须做两件事:
export TF_NEED_CUDA=1更改为export TF_NEED_CUDA=0export BUILDFLAGS="--config=cuda --copt=-m64 --linkopt=-m64 --copt=-march=native"更改为export BUILDFLAGS="--copt=-m64 --linkopt=-m64 --copt=-march=native"您还可以在--config=cuda下删除echo "Build pip package"。
在此之后,您将按照上面链接的指示进行正常操作。
这应该是可行的,但我还没有测试它自己,所以如果你有错误,请提出一个问题,在深水的Github问题跟踪器。
https://stackoverflow.com/questions/44137342
复制相似问题