tensorflow build from source

mariux64 · Mar 11, 2020 · bb877fc · bb877fc
1 parent 10307f6
commit bb877fc
Showing 1 changed file with 62 additions and 1 deletion.
diff --git a/python-3.7.6-0.build.sh b/python-3.7.6-0.build.sh
@@ -217,7 +217,6 @@ pip3 install --prefix=$PREFIX louvain        # monocle3 prep for Helene
 pip3 install --prefix=$PREFIX selenium       # 19.7.19 Kreitler  -> https://webxray.org/
 pip3 install --prefix=$PREFIX cooltools      # Maria Valieva 17.7.2019
 pip3 install --prefix=$PREFIX gmpy           # From viper context (Helpdesk Laura Glaser 9.8.2019)
-pip3 install --prefix=$PREFIX tensorflow==2.0.1         # tensorflow-2.1.0 needs spezial libs and nightly version(2.2.0) dont
 pip3 install --prefix=$PREFIX torch
 pip3 install --prefix=$PREFIX torchvision
 pip3 install --prefix=$PREFIX plotly
@@ -324,6 +323,68 @@ pip3 install --prefix=$PREFIX scour           # https://github.molgen.mpg.de/mar
 
 	python3 setup.py install --prefix $PREFIX
 )
+
+#tensorflow
+(
+	TFVERSION=2.1.0
+	TFSRC=https://github.com/tensorflow/tensorflow/archive/v${TFVERSION}/tensorflow-${TFVERSION}.tar.gz
+
+
+	. ${PREFIX}/profile
+	. /pkg/bazel-0.28.1-0/profile
+	cd $BUILDDIR
+	test -e tensorflow-${TFVERSION}.tar.gz || wget $TFSRC
+	test -d tensorflow-${TFVERSION} || tar -xf tensorflow-${TFVERSION}.tar.gz
+
+	cd tensorflow-${TFVERSION}
+#vars for config
+	export PYTHON_BIN_PATH=${PREFIX}/bin/python3
+	export USE_DEFAULT_PYTHON_LIB_PATH=1
+	export TF_NEED_JEMALLOC=1
+	export TF_NEED_KAFKA=0
+	export TF_NEED_OPENCL_SYCL=0
+	export TF_NEED_AWS=0
+	export TF_NEED_GCP=0
+	export TF_NEED_HDFS=0
+	export TF_NEED_S3=0
+	export TF_ENABLE_XLA=1
+	export TF_NEED_GDR=0
+	export TF_NEED_VERBS=0
+	export TF_NEED_OPENCL=0
+	export TF_NEED_MPI=0
+	export TF_NEED_TENSORRT=1
+	export TF_NEED_NGRAPH=0
+	export TF_NEED_IGNITE=0
+	export TF_NEED_ROCM=0
+	export TF_SET_ANDROID_WORKSPACE=0
+	export TF_DOWNLOAD_CLANG=0
+	export TF_NCCL_VERSION=2.5
+	export TF_IGNORE_MAX_BAZEL_VERSION=1
+	export NCCL_INSTALL_PATH=${CUDADIR}
+	export GCC_HOST_COMPILER_PATH=/usr/bin/gcc
+	export TF_CUDA_CLANG=0
+	export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang
+	export TF_CUDA_PATHS=${CUDADIR},${CUDADIR}/extras/CUPTI,${CUDADIR}/nvvm
+	export TF_CUDA_VERSION=$(${CUDADIR}/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p')
+	export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' ${CUDADIR}/include/cudnn.h)
+	export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.0,5.2,5.3,6.0,6.1,6.2,7.0,7.2,7.5
+	export CC_OPT_FLAGS="-march=native"
+	export TF_NEED_CUDA=1
+
+	./configure
+	bazel build \
+		//tensorflow:libtensorflow.so \
+		//tensorflow:libtensorflow_cc.so \
+		//tensorflow:install_headers \
+		//tensorflow/tools/pip_package:build_pip_package
+
+		mkdir tf-src
+		./bazel-bin/tensorflow/tools/pip_package/build_pip_package -src ./tf-src
+		cd tf-src
+		python3 setup.py build
+		python3 setup.py install -- prefix $PREFIX
+
+)
 # try to load all packages, maybe we can see installation errors
 python3 -c 'help("modules")' > /dev/null