blob: 5caee5fd5b79e47887e7c741a8eaa75716a0823b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
# Maintainer: Adria Arrufat <adria.arrufat+AUR@protonmail.ch>
# Contributor: Sven-Hendrik Haase <sh@lutzhaase.com>
pkgbase=tensorflow-git
pkgname=(tensorflow-git tensorflow-cuda-git python-tensorflow-git python-tensorflow-cuda-git)
pkgver=1.5.0+2146+g283f03c825
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 gcc6 cuda cudnn python-pip python-wheel python-setuptools)
optdepends=('cuda: GPU support'
'cudnn: GPU support')
source=("git+https://github.com/tensorflow/tensorflow")
md5sums=('SKIP')
pkgver() {
cd ${srcdir}/tensorflow
git describe --tags | sed 's/-/+/g;s/v//;'
}
prepare() {
[ -d ${srcdir}/tensorflow-cuda ] && rm -rf ${srcdir}/tensorflow-cuda
cp -r ${srcdir}/tensorflow ${srcdir}/tensorflow-cuda
# These environment variables influence the behavior of the configure call below.
export PYTHON_BIN_PATH=/usr/bin/python
export USE_DEFAULT_PYTHON_LIB_PATH=1
export CC_OPT_FLAGS="-march=native"
export TF_NEED_JEMALLOC=1
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
export TF_NEED_S3=0
export TF_NEED_OPENCL_SYCL=0
export TF_NEED_KAFKA=0
export TF_NEED_TENSORRT=0
export TF_SET_ANDROID_WORKSPACE=0
# make sure the proxy variables are in all caps, otherwise bazel ignores them
export HTTP_PROXY=`echo $http_proxy | sed -e 's/\/$//'`
export HTTPS_PROXY=`echo $https_proxy | sed -e 's/\/$//'`
}
build() {
_bazel_09_fix="--incompatible_load_argument_is_label=false"
cd ${srcdir}/tensorflow
export TF_NEED_CUDA=0
./configure
bazel build --config=opt --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix}
bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmp
cd ${srcdir}/tensorflow-cuda
export TF_NEED_CUDA=1
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/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,6.2
./configure
bazel build --config=opt --config=cuda --jobs=`nproc` //tensorflow:libtensorflow.so //tensorflow/tools/pip_package:build_pip_package ${_bazel_09_fix}
bazel-bin/tensorflow/tools/pip_package/build_pip_package ${srcdir}/tmpcuda
}
package_tensorflow-git() {
conflicts=(tensorflow)
provides=(tensorflow)
cd ${srcdir}/tensorflow
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
}
package_tensorflow-cuda-git() {
depends=(cuda cudnn)
conflicts=(tensorflow)
provides=(tensorflow)
cd ${srcdir}/tensorflow-cuda
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
}
package_python-tensorflow-git() {
depends=(python python-protobuf)
conflicts=(python-tensorflow)
provides=(python-tensorflow)
optdepends=('python-werkzeug: for using tensorboard')
cd ${srcdir}/tensorflow
WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensor*.whl")
pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies
rm -rf ${pkgdir}/usr/bin/tensorboard
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/python-tensorflow/LICENSE
}
package_python-tensorflow-cuda-git() {
depends=(python cuda cudnn python-pycuda python-protobuf)
conflicts=(python-tensorflow)
provides=(python-tensorflow)
optdepends=('python-werkzeug: for using tensorboard')
cd ${srcdir}/tensorflow-cuda
WHEEL_PACKAGE=$(find ${srcdir}/tmpcuda -name "tensor*.whl")
pip install --ignore-installed --upgrade --root $pkgdir/ $WHEEL_PACKAGE --no-dependencies
rm -rf ${pkgdir}/usr/bin/tensorboard
install -Dm644 LICENSE ${pkgdir}/usr/share/licenses/python-tensorflow/LICENSE
}
# vim:set ts=2 sw=2 et:
|