diff options
author | Daniel Bermond | 2018-07-29 16:02:32 +0000 |
---|---|---|
committer | Daniel Bermond | 2018-07-29 16:02:32 +0000 |
commit | f93ef6ae192884dd471c10fb043b324e4b4d9106 (patch) | |
tree | 8ae5f712dc9e82876950a84a33ffed68d6ad82d2 | |
download | aur-f93ef6ae192884dd471c10fb043b324e4b4d9106.tar.gz |
Initial commit
-rw-r--r-- | .SRCINFO | 18 | ||||
-rw-r--r-- | PKGBUILD | 89 |
2 files changed, 107 insertions, 0 deletions
diff --git a/.SRCINFO b/.SRCINFO new file mode 100644 index 000000000000..7ddbe954b20f --- /dev/null +++ b/.SRCINFO @@ -0,0 +1,18 @@ +pkgbase = tensorrt + pkgdesc = A high-performance deep learning inference optimizer and runtime for deep learning applications (needs registration at upstream URL and manual download) + pkgver = 4.0.1.6 + pkgrel = 1 + url = https://developer.nvidia.com/tensorrt/ + arch = x86_64 + license = custom + makedepends = poppler + makedepends = unzip + depends = cudnn + depends = python2 + depends = python + options = !strip + source = file://TensorRT-4.0.1.6.Ubuntu-16.04.4.x86_64-gnu.cuda-9.2.cudnn7.1.tar.gz + sha256sums = fc5fd2ba86225ed62e60870cc24896107770a59829363fcb49229c82bb1f4fb5 + +pkgname = tensorrt + diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..0af558930ed0 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,89 @@ +# Maintainer : Daniel Bermond < yahoo-com: danielbermond > + +# NOTE: +# In order to build the package, you need to manually download the TensorRT +# file from NVIDIA's website (registration required). Place the downloaded +# file in the PKGBUILD directory and run makepkg. +# Download website: +# https://developer.nvidia.com/tensorrt/ + +_srcname=TensorRT +_cudaver=9.2 +_cudnnver=7.1 +_graphsurgeonver=0.2.0 +_uffver=0.4.0 +_ubuntuver=16.04.4 +_ubpy3ver=35 + +pkgname=tensorrt +pkgver=4.0.1.6 +pkgrel=1 +pkgdesc='A high-performance deep learning inference optimizer and runtime for deep learning applications (needs registration at upstream URL and manual download)' +arch=('x86_64') +url='https://developer.nvidia.com/tensorrt/' +license=('custom') +depends=('cudnn' 'python2' 'python') +makedepends=('poppler' 'unzip') +options=('!strip') +source=("file://${_srcname}-${pkgver}.Ubuntu-${_ubuntuver}.${CARCH}-gnu.cuda-${_cudaver}.cudnn${_cudnnver}.tar.gz") +sha256sums=('fc5fd2ba86225ed62e60870cc24896107770a59829363fcb49229c82bb1f4fb5') + +prepare() { + cd "${_srcname}-${pkgver}/python" + + mkdir -p python{2,3} + + cd python2 + unzip -qq "../${pkgname}-${pkgver}-cp27-cp27mu-linux_${CARCH}.whl" + + cd ../python3 + unzip -qq "../${pkgname}-${pkgver}-cp${_ubpy3ver}-cp${_ubpy3ver}m-linux_${CARCH}.whl" + + cd ../../graphsurgeon + unzip -qq "graphsurgeon-${_graphsurgeonver}-py2.py3-none-any.whl" + + cd ../uff + unzip -qq "uff-${_uffver}-py2.py3-none-any.whl" + + cd ../doc + pdftotext -layout TensorRT-License.pdf +} + +package() { + cd "${_srcname}-${pkgver}" + + local _pythonver="$(python --version | sed 's/^Python[[:space:]]//' | grep -o '^[0-9]*\.[0-9]*')" + + mkdir -p "${pkgdir}/usr/lib/python"{2.7,"${_pythonver}"} + + # binaries + install -D -m755 bin/* -t "${pkgdir}/usr/bin" + + # headers + install -D -m644 include/* -t "${pkgdir}/usr/include" + + # libraries + cp -af lib/* "${pkgdir}/usr/lib" + + # python 2 + cp -af python/python2/tensorrt "${pkgdir}/usr/lib/python2.7" + + # python 3 + cp -af python/python3/tensorrt "${pkgdir}/usr/lib/python${_pythonver}" + + # graphsurgeon + cp -af graphsurgeon/graphsurgeon "${pkgdir}/usr/lib/python2.7" + cp -af graphsurgeon/graphsurgeon "${pkgdir}/usr/lib/python${_pythonver}" + + # uff + cp -af uff/uff "${pkgdir}/usr/lib/python2.7" + cp -af uff/uff "${pkgdir}/usr/lib/python${_pythonver}" + + # documentation + install -D -m644 doc/TensorRT-Developer-Guide.pdf -t "${pkgdir}/usr/share/doc/${pkgname}" + cp -af doc/{common,graphics,html} "${pkgdir}/usr/share/doc/${pkgname}" + + # license + install -D -m644 doc/TensorRT-License.txt "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE" + install -D -m644 doc/Acknowledgements.txt "${pkgdir}/usr/share/licenses/${pkgname}/ACKNOWLEDGEMENTS" +} |