summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
authorAdriĆ  Arrufat2017-10-18 20:16:24 +0200
committerAdriĆ  Arrufat2017-10-18 20:16:24 +0200
commitf6b6c92937b68ad8b880c814088b7392e3ca70da (patch)
treeaeace6cae597da88e6e667a4d23c7fc3c078577f /PKGBUILD
parent5774fea3362e3293a9ee4a998fd6524e7c3ec764 (diff)
downloadaur-f6b6c92937b68ad8b880c814088b7392e3ca70da.tar.gz
update to latest revision
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD30
1 files changed, 16 insertions, 14 deletions
diff --git a/PKGBUILD b/PKGBUILD
index d8751c58c7d..bff5b99541f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,13 +3,13 @@
pkgbase=tensorflow-git
pkgname=(tensorflow-git tensorflow-cuda-git python-tensorflow-git python-tensorflow-cuda-git)
-pkgver=1.2.0+1366+gaf8f6353a3
+pkgver=1.3.0+rc1+3547+gbedfe8ac14
pkgrel=1
pkgdesc="Library for computation using data flow graphs for scalable machine learning"
url="https://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 gcc6 cuda cudnn python-pip python-wheel python-setuptools)
optdepends=('cuda: GPU support'
'cudnn: GPU support')
source=("git+https://github.com/tensorflow/tensorflow")
@@ -31,10 +31,11 @@ prepare() {
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_MKL=0
export TF_NEED_MPI=0
+ export TF_NEED_S3=0
# make sure the proxy variables are in all caps, otherwise bazel ignores them
export HTTP_PROXY=`echo $http_proxy | sed -e 's/\/$//'`
@@ -52,14 +53,14 @@ build() {
cd ${srcdir}/tensorflow-cuda
export TF_NEED_CUDA=1
- export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-5
+ export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-6
export TF_CUDA_CLANG=0
- export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang
+ # 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 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
bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcuda
@@ -71,11 +72,11 @@ package_tensorflow-git() {
cd ${srcdir}/tensorflow
- ${srcdir}/tensorflow/tensorflow/c/generate-pc.sh --prefix=/usr --version=${pkgver:0:5}
+ tensorflow/c/generate-pc.sh --prefix=/usr --version=${pkgver}
+ install -Dm644 tensorflow.pc ${pkgdir}/usr/lib/pkgconfig/tensorflow.pc
install -Dm755 bazel-bin/tensorflow/libtensorflow.so ${pkgdir}/usr/lib/libtensorflow.so
install -Dm644 tensorflow/c/c_api.h ${pkgdir}/usr/include/tensorflow/c/c_api.h
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/tensorflow/LICENSE
- install -Dm644 tensorflow.pc ${pkgdir}/usr/lib/pkgconfig/tensorflow.pc
}
package_tensorflow-cuda-git() {
@@ -85,11 +86,11 @@ package_tensorflow-cuda-git() {
cd ${srcdir}/tensorflow-cuda
- ${srcdir}/tensorflow/tensorflow/c/generate-pc.sh --prefix=/usr --version=${pkgver:0:5}
+ tensorflow/c/generate-pc.sh --prefix=/usr --version=${pkgver}
+ install -Dm644 tensorflow.pc ${pkgdir}/usr/lib/pkgconfig/tensorflow.pc
install -Dm755 bazel-bin/tensorflow/libtensorflow.so ${pkgdir}/usr/lib/libtensorflow.so
install -Dm644 tensorflow/c/c_api.h ${pkgdir}/usr/include/tensorflow/c/c_api.h
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/tensorflow/LICENSE
- install -Dm644 tensorflow.pc ${pkgdir}/usr/lib/pkgconfig/tensorflow.pc
}
package_python-tensorflow-git() {
@@ -100,9 +101,9 @@ package_python-tensorflow-git() {
cd ${srcdir}/tensorflow
- WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensorflow-${pkgver:0:5}*.whl")
+ WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensor*.whl")
pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies
- find ${pkgdir} -name __pycache__ -exec rm -r {} +
+ rm -rf ${pkgdir}/usr/bin/tensorboard
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/python-tensorflow/LICENSE
}
@@ -115,10 +116,11 @@ package_python-tensorflow-cuda-git() {
cd ${srcdir}/tensorflow-cuda
- WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensorflow-${pkgver:0:5}*.whl")
+ WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensor*.whl")
pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies
- find ${pkgdir} -name __pycache__ -exec rm -r {} +
+ rm -rf ${pkgdir}/usr/bin/tensorboard
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/python-tensorflow/LICENSE
}
+
# vim:set ts=2 sw=2 et: