diff --git a/setup.py b/setup.py index 49a8b1e8a..addf3d567 100644 --- a/setup.py +++ b/setup.py @@ -2,6 +2,7 @@ import distutils.ccompiler import os from pathlib import Path +import platform from setuptools import setup, find_packages, Extension import subprocess import sys @@ -24,11 +25,15 @@ 'scikit-image', 'scikit-learn', 'scipy', - 'tbb', 'tqdm', 'xxhash', ] +if platform.machine() == 'x86_64': + # tbb is an optional dependency, but it is nice to have + # Add it to the requirements automatically for intel computers. + install_reqs.append('tbb') + # This will determine which compiler is being used to build the C modules compiler_type = distutils.ccompiler.get_default_compiler() if compiler_type in ("unix", "mingw32"):