[arch-commits] Commit in tensorflow/trunk (PKGBUILD)

[Sven-Hendrik Haase]

    Date: Thursday, October 5, 2017 @ 13:42:49
  Author: svenstaro
Revision: 261693

upgpkg: tensorflow 1.4.0a.git1-1

Modified:
  tensorflow/trunk/PKGBUILD

----------+
 PKGBUILD |    8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

Modified: PKGBUILD
===================================================================
--- PKGBUILD	2017-10-05 13:29:52 UTC (rev 261692)
+++ PKGBUILD	2017-10-05 13:42:49 UTC (rev 261693)
@@ -4,9 +4,9 @@
 
 pkgbase=tensorflow
 pkgname=(tensorflow tensorflow-cuda python-tensorflow python-tensorflow-cuda)
-# I went with git for just this release because I couldn't be bothered to find all
-# patches needed for bazel 0.6. It's more than the issue tracker would have you think!
-pkgver=1.4.0.git1
+# I went with git for just this release because I couldn't be bothered to
+# backport all the patches needed for bazel 0.6, cuda 9, and cudnn 7.
+pkgver=1.4.0a.git1
 pkgrel=1
 pkgdesc="Library for computation using data flow graphs for scalable machine learning"
 url="https://www.tensorflow.org/"
@@ -57,7 +57,7 @@
   export TF_CUDA_VERSION=$($CUDA_TOOLKIT_PATH/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p')
   export CUDNN_INSTALL_PATH=/opt/cuda
   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
+  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