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

[Sven-Hendrik Haase]

    Date: Tuesday, August 8, 2017 @ 21:08:51
  Author: svenstaro
Revision: 249619

Use cudnn6 instead of cudnn

Modified:
  tensorflow/trunk/PKGBUILD

----------+
 PKGBUILD |   10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

Modified: PKGBUILD
===================================================================
--- PKGBUILD	2017-08-08 21:05:55 UTC (rev 249618)
+++ PKGBUILD	2017-08-08 21:08:51 UTC (rev 249619)
@@ -5,12 +5,12 @@
 pkgbase=tensorflow
 pkgname=(tensorflow tensorflow-cuda python-tensorflow python-tensorflow-cuda)
 pkgver=1.2.1
-pkgrel=4
+pkgrel=5
 pkgdesc="Library for computation using data flow graphs for scalable machine learning"
 url="https://www.tensorflow.org/"
 license=('Apache2')
 arch=('x86_64')
-makedepends=('git' 'bazel' 'python-numpy' 'gcc5' 'cuda' 'cudnn' 'python-pip' 'python-wheel' 'python-setuptools')
+makedepends=('git' 'bazel' 'python-numpy' 'gcc5' 'cuda' 'cudnn6' 'python-pip' 'python-wheel' 'python-setuptools')
 source=("https://github.com/tensorflow/tensorflow/archive/v${pkgver}.tar.gz"
         10868.patch
         11174.patch
@@ -59,7 +59,7 @@
   # export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang
   export CUDA_TOOLKIT_PATH=/opt/cuda
   export TF_CUDA_VERSION=$($CUDA_TOOLKIT_PATH/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p')
-  export CUDNN_INSTALL_PATH=/opt/cuda
+  export CUDNN_INSTALL_PATH=/opt/cudnn6
   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
 
@@ -79,7 +79,7 @@
 }
 
 package_tensorflow-cuda() {
-  depends=(cuda cudnn)
+  depends=(cuda cudnn6)
   conflicts=(tensorflow)
   provides=(tensorflow)
 
@@ -109,7 +109,7 @@
 }
 
 package_python-tensorflow-cuda() {
-  depends=(python cuda cudnn python-pycuda python-protobuf)
+  depends=(python cuda cudnn6 python-pycuda python-protobuf)
   conflicts=(python-tensorflow)
   provides=(python-tensorflow)
   optdepends=('python-werkzeug: for using tensorboard'