diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 30 |
1 files changed, 16 insertions, 14 deletions
@@ -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: |