2017-05-23 14 views
1

GPUを搭載したマシンでTensorFlowバックエンドでモデルを構築しました。 今、GPUを搭載していないプロダクションマシンでこのモデルを使いたいです。 TensorFlowバックエンドがインストールされていないこのモデルは使用できませんが、TensorFlowバックエンドはGPUなしのマシンにはインストールできません。CUDAなしでh2oでTensorFlowを構築する

GPUをサポートしていないH2OのTensorFlowバックエンドを構築するオプションはありますか?

答えて

3

が、私はまた、GPU

ないマシン上でTensorFlowバックエンドをインストールすることはできません、あなたがしようとしたエラーを得たわけですか?

GPUがサポートされていないH2O用のTensorFlowバックエンドを構築するオプションはありますか?

はい、お使いのマシンによっては、簡単かもしれないし、ちょっとハッキリかもしれません。あなたが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を除去することがあります:あなたは2つのことを行う必要があるだろう。

この後は、上記の通常の手順に従います。

これは動作するはずですが、自分でテストしていないため、エラーが発生した場合はDeepWaterのGithub問題トラッカーで問題を提起してください。

+0

現在作業中です。ありがとうございました :) –

関連する問題