tensorflow 빌드하기 – mac

By | 2017년 5월 31일

필요한 프로그램 설치

brew install bazel

필요한 python 라이브러리 설치

 pip3 install numpy wheel

소스 다운로드 및 빌드

git clone --recursive https://github.com/tensorflow/tensorflow

cd tensorflow

./configure

python 용 pip 패키지 만들기

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-msse4.2 //tensorflow/tools/pip_package:build_pip_package

(GPU를 사용하려면 –config=cuda 추가)

–copt부분은 cpu가 지원되는 것에 따라 약간씩 다를 수 있음

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

pip 패키지 빌드가 끝나면 pip 패키지 설치

pip install /tmp/tensorflow_pkg/tensorflow-1.1.0-cp36-cp36m-macosx_10_6_intel.whl

 

Share this:

2 thoughts on “tensorflow 빌드하기 – mac

댓글 남기기

이메일은 공개되지 않습니다. 필수 입력창은 * 로 표시되어 있습니다