Date: Tuesday, October 3, 2017 @ 10:21:30 Author: svenstaro Revision: 261370
Prepare for cuda9 Modified: tensorflow/trunk/PKGBUILD ----------+ PKGBUILD | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2017-10-03 10:20:46 UTC (rev 261369) +++ PKGBUILD 2017-10-03 10:21:30 UTC (rev 261370) @@ -4,23 +4,22 @@ pkgbase=tensorflow pkgname=(tensorflow tensorflow-cuda python-tensorflow python-tensorflow-cuda) -pkgver=1.3.0 -pkgrel=5 +# 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 +pkgrel=1 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' - 'cudnn6' 'python-pip' 'python-wheel' 'python-setuptools') +makedepends=('git' 'bazel' 'python-numpy' 'gcc6' 'cuda' + 'cudnn' 'python-pip' 'python-wheel' 'python-setuptools') optdepends=('tensorboard: Tensorflow visualization toolkit') -source=("https://github.com/tensorflow/tensorflow/archive/v${pkgver}.tar.gz";) -sha512sums=('b74228a13537e4cbcf8781167480a0bccb2c1ecf7bf21a02e35a182e073395f919645205036fd2193d5a186cfa8a869ba32c04fa5bd8f82cbe0ad8e56926398b') +source=("git://github.com/tensorflow/tensorflow.git#commit=64f0ebd") +sha512sums=('SKIP') prepare() { - # Necessary because of https://github.com/tensorflow/tensorflow/issues/12979#event-1250326815 - cd tensorflow-${pkgver} - sed -ri "/^\W+sha256 = \"[^\"]+\"\W+$/d" tensorflow/workspace.bzl - cd .. + cp -r tensorflow tensorflow-${pkgver} cp -r tensorflow-${pkgver} tensorflow-${pkgver}-cuda # These environment variables influence the behavior of the configure call below. @@ -32,6 +31,7 @@ export TF_NEED_GCP=0 export TF_NEED_HDFS=0 export TF_ENABLE_XLA=1 + export TF_NEED_GDR=0 export TF_NEED_VERBS=0 export TF_NEED_OPENCL=0 export TF_NEED_MPI=0 @@ -49,13 +49,13 @@ cd ${srcdir}/tensorflow-${pkgver}-cuda export TF_NEED_CUDA=1 - export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-5 - # For next version instead of the gcc-5 stuff: + # For next version instead of the gcc-6 stuff: + export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-6 export TF_CUDA_CLANG=0 # 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/cudnn6 + 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 @@ -75,7 +75,7 @@ } package_tensorflow-cuda() { - depends=(cuda cudnn6) + depends=(cuda cudnn) conflicts=(tensorflow) provides=(tensorflow) @@ -105,7 +105,7 @@ } package_python-tensorflow-cuda() { - depends=(python cuda cudnn6 python-pycuda python-protobuf) + depends=(python cuda cudnn python-pycuda python-protobuf) conflicts=(python-tensorflow) provides=(python-tensorflow)
