yaroslavvb / tensorflow-community-wheels

Place to upload links to TensorFlow wheels
452 stars 35 forks source link

TensorFlow 2.6.2 No AVX, No GPU, Python 3.7, 3.8, 3.9, Ubuntu 18.04, multiple Archs #204

Open Novaal opened 2 years ago

Novaal commented 2 years ago

Built on Ubuntu 18.04. Builds are not tested and provided as is.

Example configuration for Python 3.7 and westmere:

PYTHON_VERSION=python3.7
PYTHON_BIN_PATH=$(which $PYTHON_VERSION) \
PYTHON_LIB_PATH=$($PYTHON_VERSION -c "import pip; print(pip.__path__[0].rstrip('/pip'))") \
TF_NEED_CUDA=0 \
TF_NEED_ROCM=0 \
TF_DOWNLOAD_CLANG=0 \
CC_OPT_FLAGS="-march=westmere -Wno-sign-compare" \
TF_SET_ANDROID_WORKSPACE=0 \
TF_ENABLE_XLA=1 \
TF_NEED_OPENCL_SYCL=0 \
TF_NEED_MPI=0 \
./configure

This produces the following .tf_configure.bazelrc for Python 3.7.

build --action_env PYTHON_BIN_PATH="/usr/bin/python3.7"
build --action_env PYTHON_LIB_PATH="/usr/local/lib/python3.7/dist-packages"
build --python_path="/usr/bin/python3.7"
build --define=with_xla_support=true
build:opt --copt=-march=westmere
build:opt --host_copt=-march=westmere
build:opt --copt=-Wno-sign-compare
build:opt --host_copt=-Wno-sign-compare
test --flaky_test_attempts=3
test --test_size_filters=small,medium
test:v1 --test_tag_filters=-benchmark-test,-no_oss,-gpu,-oss_serial
test:v1 --build_tag_filters=-benchmark-test,-no_oss,-gpu
test:v2 --test_tag_filters=-benchmark-test,-no_oss,-gpu,-oss_serial,-v1only
test:v2 --build_tag_filters=-benchmark-test,-no_oss,-gpu,-v1only
GCC Compiler Option
(-march)
Python 3.7 Python 3.8 Python 3.9 All Builds
barcelona Download Download Download Link
btver1 Download Download Download Link
core2 Download Download Download Link
nehalem Download Download Download Link
westmere Download Download Download Link

Install with:

pip install --ignore-installed --upgrade tensorflow-2.6.2-cp37-cp37m-linux_x86_64.whl

Please let me know here if it works for you.

cryoff commented 2 years ago

Worked for me for workstation with these processors: https://ark.intel.com/content/www/us/en/ark/products/47920/intel-xeon-processor-x5670-12m-cache-2-93-ghz-6-40-gts-intel-qpi.html