summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD18
1 files changed, 6 insertions, 12 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 3a2c889bbb4c..8149ed16e31e 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -4,25 +4,19 @@
pkgname=libtorch-cuda
_name="${pkgname%%-cuda}"
pkgver=2.2.2
-_cuver=121
-pkgrel=1
+_ver=cu121
+pkgrel=2
pkgdesc='PyTorch C++ API'
arch=('x86_64')
url='pytorch.org'
license=('BSD')
-depends=('cuda' 'cudnn' 'cmake' 'onednn')
provides=("${_name}")
conflicts=("${_name}")
-source=("${pkgname}-${pkgver}.zip"::"https://download.pytorch.org/libtorch/cu${_cuver}/libtorch-shared-with-deps-${pkgver}%2Bcu${_cuver}.zip")
+source=("${_name}_${_ver}-${pkgver}.zip"::"https://download.pytorch.org/libtorch/${_ver}/libtorch-shared-with-deps-${pkgver}%2B${_ver}.zip")
sha512sums=('c8fc08ec5a195292570e72f65f5ed9c240206cf8c5ed4baf3963bd5654feace476ff7787532e685a2e18a256cfb8332e7b6a20d467bbfbfc8a7c8d1137df3589')
-options=(!strip libtool staticlibs)
+options=(!strip libtool staticlibs !debug)
package() {
- rm ${srcdir}/libtorch/lib/libcudnn*
- rm ${srcdir}/libtorch/include/dnnl*
-
- install -vDm755 "${srcdir}/libtorch/lib/"* -t "${pkgdir}/usr/lib"
- install -vd "${pkgdir}/usr/include" "${pkgdir}/usr/share/cmake"
- cp -r "${srcdir}/libtorch/include/"* "${pkgdir}/usr/include/"
- cp -r "${srcdir}/libtorch/share/cmake/"* "${pkgdir}/usr/share/cmake/"
+ install -vdm755 "${pkgdir}/opt"
+ cp -r "${srcdir}/${_name}" "${pkgdir}/opt"
}