Date: Thursday, October 6, 2022 @ 08:23:58 Author: svenstaro Revision: 1321819
upgpkg: tensorflow 2.10.0-2: cuda 11.8 rebuild Modified: tensorflow/trunk/PKGBUILD ----------+ PKGBUILD | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2022-10-06 08:20:56 UTC (rev 1321818) +++ PKGBUILD 2022-10-06 08:23:58 UTC (rev 1321819) @@ -7,7 +7,7 @@ pkgname=(tensorflow tensorflow-opt tensorflow-cuda tensorflow-opt-cuda python-tensorflow python-tensorflow-opt python-tensorflow-cuda python-tensorflow-opt-cuda) pkgver=2.10.0 _pkgver=2.10.0 -pkgrel=1 +pkgrel=2 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/"; license=('APACHE') @@ -110,7 +110,8 @@ # https://github.com/tensorflow/tensorflow/blob/1ba2eb7b313c0c5001ee1683a3ec4fbae01105fd/third_party/gpus/cuda_configure.bzl#L411-L446 # according to the above, we should be specifying CUDA compute capabilities as 'sm_XX' or 'compute_XX' from now on # add latest PTX for future compatibility - export TF_CUDA_COMPUTE_CAPABILITIES=sm_52,sm_53,sm_60,sm_61,sm_62,sm_70,sm_72,sm_75,sm_80,sm_86,sm_87,compute_87 + # Valid values can be discovered from nvcc --help + export TF_CUDA_COMPUTE_CAPABILITIES=sm_52,sm_53,sm_60,sm_61,sm_62,sm_70,sm_72,sm_75,sm_80,sm_86,sm_87,sm_89,sm_90,compute_90 export BAZEL_ARGS="--config=mkl -c opt" }
