Date: Monday, June 22, 2020 @ 12:02:40 Author: svenstaro Revision: 650411
Prepare CUDA 11 build Modified: tensorflow/trunk/PKGBUILD ----------+ PKGBUILD | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2020-06-22 12:02:31 UTC (rev 650410) +++ PKGBUILD 2020-06-22 12:02:40 UTC (rev 650411) @@ -7,13 +7,13 @@ pkgname=(tensorflow tensorflow-opt tensorflow-cuda tensorflow-opt-cuda python-tensorflow python-tensorflow-opt python-tensorflow-cuda python-tensorflow-opt-cuda) pkgver=2.2.0 _pkgver=2.2.0 -pkgrel=1 +pkgrel=2 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/"; license=('APACHE') arch=('x86_64') depends=('c-ares' 'intel-mkl' 'onednn') -makedepends=('bazel' 'python-numpy' 'cuda' 'nvidia-utils' 'nccl' 'git' 'gcc' +makedepends=('bazel' 'python-numpy' 'cuda' 'nvidia-utils' 'nccl' 'git' 'gcc9' 'cudnn' 'python-pip' 'python-wheel' 'python-setuptools' 'python-h5py' 'python-keras-applications' 'python-keras-preprocessing') optdepends=('tensorboard: Tensorflow visualization toolkit' @@ -77,14 +77,19 @@ export TF_IGNORE_MAX_BAZEL_VERSION=1 export TF_MKL_ROOT=/opt/intel/mkl export NCCL_INSTALL_PATH=/usr - export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-8 - export HOST_CXX_COMPILER_PATH=/usr/bin/gcc-8 + export GCC_HOST_COMPILER_PATH=/usr/bin/gcc-9 + export HOST_C_COMPILER=/usr/bin/gcc-9 + export HOST_CXX_COMPILER=/usr/bin/g++-9 export TF_CUDA_CLANG=0 # Clang currently disabled because it's not compatible at the moment. export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang export TF_CUDA_PATHS=/opt/cuda,/usr/lib,/usr export TF_CUDA_VERSION=$(/opt/cuda/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p') export TF_CUDNN_VERSION=$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' /usr/include/cudnn.h) - export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.0,5.2,5.3,6.0,6.1,6.2,7.0,7.2,7.5 + export TF_CUDA_COMPUTE_CAPABILITIES=5.2,5.3,6.0,6.1,6.2,7.0,7.2,7.5,8.0 + + # Required until https://github.com/tensorflow/tensorflow/issues/39467 is fixed. + export CC=gcc-9 + export CXX=g++-9 } build() {
