diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 8 |
1 files changed, 5 insertions, 3 deletions
@@ -3,7 +3,7 @@ pkgbase=tensorflow-git pkgname=(tensorflow-git tensorflow-cuda-git python-tensorflow-git python-tensorflow-cuda-git) -pkgver=1.3.0+rc1+3547+gbedfe8ac14 +pkgver=1.5.0+rc1+1320+g6e6c7e9545 pkgrel=1 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://tensorflow.org/" @@ -36,6 +36,7 @@ prepare() { export TF_NEED_OPENCL=0 export TF_NEED_MPI=0 export TF_NEED_S3=0 + export TF_NEED_OPENCL_SYCL=0 # make sure the proxy variables are in all caps, otherwise bazel ignores them export HTTP_PROXY=`echo $http_proxy | sed -e 's/\/$//'` @@ -43,12 +44,13 @@ prepare() { } build() { + _bazel_09_fix="--incompatible_load_argument_is_label=false" cd ${srcdir}/tensorflow export TF_NEED_CUDA=0 ./configure - bazel build --config=opt //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + bazel build --config=opt --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix} bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmp cd ${srcdir}/tensorflow-cuda @@ -62,7 +64,7 @@ build() { export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' $CUDNN_INSTALL_PATH/include/cudnn.h) export TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.1,6.2 ./configure - bazel build --config=opt --config=cuda //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package + bazel build --config=opt --config=cuda --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix} bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcuda } |