2017-05-23 6 views
1

GPU가있는 머신에서 TensorFlow 백엔드 모델을 구축했습니다. 이제 GPU가없는 프로덕션 머신에서이 모델을 사용하고 싶습니다. TensorFlow 백엔드가 설치되어 있지 않으면이 모델을 사용할 수 없지만 GPU가없는 컴퓨터에는 TensorFlow 백엔드를 설치할 수 없습니다.CUDA없이 h2o로 TensorFlow 빌드하기

GPU가 지원되지 않는 H2O 용 TensorFlow 백엔드를 빌드 할 수있는 옵션이 있습니까?

답변

3

그러나 나는 또한

GPU

없이 컴퓨터에 TensorFlow 백엔드를 설치할 수 없습니다 당신이 시도하고 오류가있어 의미합니까?

GPU가 지원되지 않는 H2O 용 TensorFlow 백엔드를 빌드 할 수있는 옵션이 있습니까?

예. 컴퓨터에 따라 쉽지 않을 수도 있습니다. MacOS 시스템에서 실행중인 경우 즉시 사용할 수 있습니다. follow our instruction.

Linux 기반 시스템을 사용하는 경우 먼저 스크립트를 변경해야합니다 (정확히 this one). 당신은 두 가지 일을해야 할 것 :

  • 변화 또한 echo "Build pip package"에서 --config=cuda을 제거 할 수 export TF_NEED_CUDA=1export BUILDFLAGS="--config=cuda --copt=-m64 --linkopt=-m64 --copt=-march=native"

export BUILDFLAGS="--copt=-m64 --linkopt=-m64 --copt=-march=native"로 변경

  • export TF_NEED_CUDA=0에.

    그런 다음 정상적으로 링크 된 지침을 따릅니다.

    이 방법은 효과가 있지만 직접 테스트하지는 않았으므로 오류가 발생하면 DeepWater의 Github 문제 추적기에서 문제를 제기하십시오.

  • +0

    현재 작동 중입니다. 다시 감사합니다 :) –