diff options
-rw-r--r-- | .SRCINFO | 4 | ||||
-rw-r--r-- | PKGBUILD | 6 |
2 files changed, 6 insertions, 4 deletions
@@ -1,13 +1,15 @@ pkgbase = tensorflow-rocm pkgdesc = Library for computation using data flow graphs for scalable machine learning pkgver = 2.2.0 - pkgrel = 1 + pkgrel = 2 url = https://www.tensorflow.org/ arch = x86_64 license = APACHE makedepends = bazel makedepends = python-numpy makedepends = rocm + makedepends = rocm-libs + makedepends = miopen makedepends = rccl makedepends = git makedepends = gcc9 @@ -8,14 +8,14 @@ pkgbase=tensorflow-rocm pkgname=(tensorflow-rocm tensorflow-opt-rocm python-tensorflow-rocm python-tensorflow-opt-rocm) 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' 'rocm' 'rccl' 'git' 'gcc9' - 'python-pip' 'python-wheel' 'python-setuptools' 'python-h5py' +makedepends=('bazel' 'python-numpy' 'rocm' 'rocm-libs' 'miopen' 'rccl' 'git' + 'gcc9' 'python-pip' 'python-wheel' 'python-setuptools' 'python-h5py' 'python-keras-applications' 'python-keras-preprocessing') optdepends=('tensorboard: Tensorflow visualization toolkit' 'python-pasta: tf_upgrade_v2 tool') |