TensorFlow r0.7: change pip dependency for numpy to 1.10.1 on OS X

Vijay Vasudevan · Vijay Vasudevan · commit 376bb0e94ccf · 2016-02-19T18:12:58.000-08:00
pip3 install numpy==1.8.2 results in failures to find Python.h
during our wheel building process, and it appears to be a bug
in numpy.distutils at 1.8.2.
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
@@ -19,6 +19,7 @@
 
 import fnmatch
 import os
+import platform
 import re
 import sys
 
@@ -28,8 +29,15 @@
 
 _VERSION = '0.7.1'
 
+numpy_version = "1.8.2"
+if platform.system() == "Darwin":
+  # There are bugs with numpy pip installation on OS X prior to
+  # 1.10.1, so on mac we require a higher version than on other
+  # platforms.
+  numpy_version = "1.10.1"
+
 REQUIRED_PACKAGES = [
-    'numpy >= 1.8.2',
+    'numpy >= %s' % numpy_version,
     'six >= 1.10.0',
     'protobuf == 3.0.0b2',
 ]
