在没有 CUDA 的情况下在 h2o 中构建 TensorFlow
Building TensorFlow in h2o without CUDA
我已经在配备 GPU 的机器上使用 TensorFlow 后端构建了模型。
现在我想在没有 GPU 的生产机器上使用这个模型。
我无法在没有安装 TensorFlow 后端的情况下使用此模型,但我也无法在没有 GPU 的机器上安装 TensorFlow 后端。
是否有任何选项可以在没有 GPU 支持的情况下为 H2O 构建 TensorFlow 后端?
but I also cannot install TensorFlow backend on machine without GPU
你的意思是你尝试了却出错了?
Is there any option to built TensorFlow backend for H2O without GPU support?
是的,但根据您的机器,这可能很简单,也可能有点棘手。如果您 运行 在 MacOS 系统上,那么它应该开箱即用,只需 follow our instruction.
如果您使用的是基于 Linux 的系统,那么您必须首先更改我们的脚本,确切地说是 this one。你必须做两件事:
- 将
export TF_NEED_CUDA=1
更改为 export TF_NEED_CUDA=0
- 将
export BUILDFLAGS="--config=cuda --copt=-m64 --linkopt=-m64 --copt=-march=native"
更改为export BUILDFLAGS="--copt=-m64 --linkopt=-m64 --copt=-march=native"
您也可以删除 echo "Build pip package"
下的 --config=cuda
。
在此之后,您可以照常按照上面链接的说明进行操作。
这应该可行,但我还没有亲自测试过,所以如果您遇到错误,请在 DeepWater 的 Github 问题跟踪器上提出问题。
我已经在配备 GPU 的机器上使用 TensorFlow 后端构建了模型。 现在我想在没有 GPU 的生产机器上使用这个模型。 我无法在没有安装 TensorFlow 后端的情况下使用此模型,但我也无法在没有 GPU 的机器上安装 TensorFlow 后端。
是否有任何选项可以在没有 GPU 支持的情况下为 H2O 构建 TensorFlow 后端?
but I also cannot install TensorFlow backend on machine without GPU
你的意思是你尝试了却出错了?
Is there any option to built TensorFlow backend for H2O without GPU support?
是的,但根据您的机器,这可能很简单,也可能有点棘手。如果您 运行 在 MacOS 系统上,那么它应该开箱即用,只需 follow our instruction.
如果您使用的是基于 Linux 的系统,那么您必须首先更改我们的脚本,确切地说是 this one。你必须做两件事:
- 将
export TF_NEED_CUDA=1
更改为export TF_NEED_CUDA=0
- 将
export BUILDFLAGS="--config=cuda --copt=-m64 --linkopt=-m64 --copt=-march=native"
更改为export BUILDFLAGS="--copt=-m64 --linkopt=-m64 --copt=-march=native"
您也可以删除 echo "Build pip package"
下的 --config=cuda
。
在此之后,您可以照常按照上面链接的说明进行操作。
这应该可行,但我还没有亲自测试过,所以如果您遇到错误,请在 DeepWater 的 Github 问题跟踪器上提出问题。