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'
