[arch-commits] Commit in tensorflow/trunk (PKGBUILD)
Sven-Hendrik Haase
svenstaro at archlinux.org
Sun Dec 9 07:14:55 UTC 2018
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
}
More information about the arch-commits
mailing list