diff options
author | Daniel Bermond | 2021-11-03 16:29:45 -0300 |
---|---|---|
committer | Daniel Bermond | 2021-11-03 16:29:45 -0300 |
commit | ae8f14ec73a4c80308b9f5d0776af01857f1284e (patch) | |
tree | 58a40adb76dd8a565e6173ba6400aa73973e3ad8 | |
parent | bd137837b77b99d46d55f455868ac42a362ab5a9 (diff) | |
download | aur-ae8f14ec73a4c80308b9f5d0776af01857f1284e.tar.gz |
Add the pytorch extension
-rw-r--r-- | .SRCINFO | 4 | ||||
-rw-r--r-- | PKGBUILD | 17 |
2 files changed, 12 insertions, 9 deletions
@@ -1,6 +1,6 @@ pkgbase = nvidia-vpf-git pkgdesc = NVIDIA Video Processing Framework (git version) - pkgver = 1.0.r1.gba47dca + pkgver = 1.1.r14.g5e951e4 pkgrel = 1 url = https://github.com/NVIDIA/VideoProcessingFramework/ arch = x86_64 @@ -8,10 +8,12 @@ pkgbase = nvidia-vpf-git makedepends = git makedepends = cmake makedepends = nvidia-sdk + makedepends = python-setuptools depends = cuda depends = nvidia-utils depends = ffmpeg depends = python + depends = python-pytorch-cuda provides = nvidia-vpf conflicts = nvidia-vpf options = !emptydirs @@ -1,14 +1,14 @@ # Maintainer: Daniel Bermond <dbermond@archlinux.org> pkgname=nvidia-vpf-git -pkgver=1.0.r1.gba47dca +pkgver=1.1.r14.g5e951e4 pkgrel=1 pkgdesc='NVIDIA Video Processing Framework (git version)' arch=('x86_64') url='https://github.com/NVIDIA/VideoProcessingFramework/' license=('Apache') -depends=('cuda' 'nvidia-utils' 'ffmpeg' 'python') -makedepends=('git' 'cmake' 'nvidia-sdk') +depends=('cuda' 'nvidia-utils' 'ffmpeg' 'python' 'python-pytorch-cuda') +makedepends=('git' 'cmake' 'nvidia-sdk' 'python-setuptools') provides=('nvidia-vpf') conflicts=('nvidia-vpf') options=('!emptydirs') @@ -20,7 +20,7 @@ pkgver() { } build() { - export CUDAToolkit_ROOT='/opt/cuda' + export CUDA_HOME='/opt/cuda' export CXXFLAGS+=' -I/opt/cuda/include' export LDFLAGS+=' -L/opt/cuda/lib64' cmake -B build -S VideoProcessingFramework \ @@ -29,6 +29,7 @@ build() { -DCMAKE_CUDA_COMPILER:FILEPATH='/opt/cuda/bin/nvcc' \ -DCMAKE_SKIP_RPATH:BOOL='YES' \ -DGENERATE_PYTHON_BINDINGS:BOOL='TRUE' \ + -DGENERATE_PYTORCH_EXTENSION:BOOL='TRUE' \ -DVIDEO_CODEC_SDK_INCLUDE_DIR:PATH='/usr/include/nvidia-sdk' \ -Wno-dev make -C build @@ -37,10 +38,10 @@ build() { package() { make -C build DESTDIR="$pkgdir" install - local _pyver - _pyver="$(python -c 'import sys; print("%s.%s" %sys.version_info[0:2])')" - install -d -m755 "${pkgdir}/usr"/{"lib/python${_pyver}/site-packages",share/nvidia-vpf/samples} - mv "${pkgdir}/usr/bin"/PyNvCodec.cpython*.so "${pkgdir}/usr/lib/python${_pyver}/site-packages" + local _sitepkg + _sitepkg="$(python -c 'import site; print(site.getsitepackages()[0])')" + install -d -m755 "$pkgdir"{"$_sitepkg",/usr/share/nvidia-vpf/samples} + mv "${pkgdir}/usr/bin"/Py{,torch}NvCodec"$(python-config --extension-suffix)" "${pkgdir}${_sitepkg}" mv "${pkgdir}/usr/bin"/*.so* "${pkgdir}/usr/lib" mv "${pkgdir}/usr/bin"/*.py "${pkgdir}/usr/share/nvidia-vpf/samples" chmod a+x "${pkgdir}/usr/share/nvidia-vpf/samples"/*.py |