본문 바로가기

카테고리 없음

Build Tensorflow CPU for windows (C API)

** Windows, CPU 버전, C API 사용을 위한 build 입니다 **

** 2019.04.05 일자 tensorflow master 기준입니다 **

 

참고문서: 

https://www.tensorflow.org/install/source_windows

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/lib_package/README.md#java-library

 

tensorflow/tensorflow

An Open Source Machine Learning Framework for Everyone - tensorflow/tensorflow

github.com

 

Build from source on Windows  |  TensorFlow

Build a TensorFlow pip package from source and install it on Windows. Note: We already provide well-tested, pre-built TensorFlow packages for Windows systems. Setup for Windows Install the following build tools to configure your Windows development environ

www.tensorflow.org

  1. Visual Studio 2015 or later 설치 (community 버전도 가능)
  2. 64비트 버전 Python 3.5 또는 3.6 버전 설치
  3. pip 로 필요한 라이브러리 설치
  4. Bazel 다운로드
    • https://github.com/bazelbuild/bazel/releases 에서 bazel-{version}-windows-x86_64.exe 파일 다운로드 후 bazel.exe 로 이름변경
    • 편한 디렉토리에 복사해 놓고 사용하면 됨
      예) c:\bazel\bazel.exe
  5. MSYS2 설치
  6. 환경변수 편집
    • PATH 에 bazel 경로, mysis2\usr\bin 위치 추가
      예) c:\msys64\usr\bin, c:\bazel
    • BAZEL_SH = c:\msys64\usr\bin\bash.exe
    • BAZEL_VS = c:\Program Files (x86)\Microsoft Visual Studio 14.0
  7. 추가로 필요한 utility 설치. 명령 프롬프트 (cmd.exe) 에서 아래 명령 실행
    • c:\> pacman -S git patch unzip
  8. 텐서플로우 소스코드 다운로드
  9. 빌드 환경 설정
    • cd tensorflow
    • python ./configure.py
  10. C API 빌드
    • bazel build --config=opt --copt=/arch:AVX2 //tensorflow/tools/lib_package:libtensorflow
      • AVX2 를 지원하는 CPU 가 아니라면 arch:AVX2 옵션을 제외
  11. \bazel-bin\tensorflow\tools\lib_package 폴더에 생성된 압축파일(header, lib, dll)을 사용