Date: Sunday, December 9, 2018 @ 07:14:54 Author: svenstaro Revision: 413006
cudnn 7.4 rebuild Modified: tensorflow/trunk/PKGBUILD ----------+ PKGBUILD | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2018-12-09 01:04:45 UTC (rev 413005) +++ PKGBUILD 2018-12-09 07:14:54 UTC (rev 413006) @@ -6,7 +6,7 @@ pkgname=(tensorflow tensorflow-opt tensorflow-cuda tensorflow-opt-cuda python-tensorflow python-tensorflow-opt python-tensorflow-cuda python-tensorflow-opt-cuda) pkgver=1.12.0 _pkgver=1.12.0 -pkgrel=2 +pkgrel=3 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/"; license=('APACHE') @@ -17,6 +17,8 @@ 'python-keras-applications' 'python-keras-preprocessing') optdepends=('tensorboard: Tensorflow visualization toolkit') source=("https://github.com/tensorflow/tensorflow/archive/v${_pkgver}.tar.gz"; + # Temp fix for https://github.com/tensorflow/tensorflow/issues/24124 + https://github.com/bazelbuild/bazel/releases/download/0.19.2/bazel-0.19.2-linux-x86_64 # python37.patch protobuf-version-bump.patch protobuf-python37.patch @@ -25,6 +27,7 @@ explicitly_import_bazelrc.patch 5b14577d42842871f1cb0eb8dfe77d32db1eb654.patch) sha512sums=('b145a9118856aa00a829ab6af89bff4e1e131371c96d77b07532544112803c4574d97ef224b28a64437a2af8db4286786dc0b4123efe110b2aa734b443a7e238' + 'bd87b7c8c517ff2c4783b7b09c4609e9979d79985ff8ed1d564a8f2f49ad8588bfb72eb5d20b3ecc6c7e7e20b64e79e3913c4705b14d7a330ffbc2a0e09f10c4' 'b9821d8eed1bb0125ddb8ebc46e8a4b8630789aac9b35aa4ecf24459476370f8b57fcf016952d7b7b61bee4177124abba343b7c9e956645b8895d2680f84e29a' 'c5e24f584153752e7e22f472c8b0378dce8d0e07fa44598f571d27707d1ac5d0917ca1b6556c59259ef27c1405da6f83b9080ed803550ae6a6121ea3054d3111' 'e13399375a8d61f695f67be20d86570b0e7e9b8682ea00eb689b3638e2b12ce50cb45fbe46f2ce1b1f065d043a29b8fac1b96eeec01317415ef2d8b51ac8332a' @@ -69,6 +72,8 @@ export TF_DOWNLOAD_CLANG=0 export TF_NCCL_VERSION=2.3 export NCCL_INSTALL_PATH=/usr + + chmod +x "${srcdir}"/bazel-0.19.2-linux-x86_64 } build() { @@ -76,7 +81,7 @@ export CC_OPT_FLAGS="-march=x86-64" export TF_NEED_CUDA=0 ./configure - bazel build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + "${srcdir}"/bazel-0.19.2-linux-x86_64 build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmp cd ${srcdir}/tensorflow-${_pkgver}-cuda @@ -91,7 +96,7 @@ export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' /usr/include/cudnn.h) export TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.1,6.2,7.0,7.2,7.5 ./configure - bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + "${srcdir}"/bazel-0.19.2-linux-x86_64 build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcuda cd ${srcdir}/tensorflow-${_pkgver}-opt @@ -98,7 +103,7 @@ export CC_OPT_FLAGS="-march=haswell" export TF_NEED_CUDA=0 ./configure - bazel build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + "${srcdir}"/bazel-0.19.2-linux-x86_64 build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpopt cd ${srcdir}/tensorflow-${_pkgver}-opt-cuda @@ -113,7 +118,7 @@ export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' /usr/include/cudnn.h) export TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.1,6.2,7.0,7.2,7.5 ./configure - bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + "${srcdir}"/bazel-0.19.2-linux-x86_64 build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcudaopt }
