blob: 1ae32ac2f391e0b1c696c237ebddc1d66fd468cb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
# Maintainer: Zhirui Dai <daizhirui at hotmail dot com>
_pkgname=libtorch
pkgbase=libtorch-cxx11abi
pkgname=(
libtorch-cxx11abi-cuda
libtorch-cxx11abi-rocm
libtorch-cxx11abi-cpu
)
pkgver="2.3.0"
_cuda_version="cu121"
_rocm_version="rocm5.7"
pkgrel=1
_pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CXX11 ABI)"
pkgdesc="${_pkgdesc}"
arch=('x86_64')
url="https://pytorch.org"
license=('BSD')
depends=(pybind11)
makedepends=()
source=(
"${_pkgname}-${pkgver}-cuda.zip::https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-${pkgver}%2B${_cuda_version}.zip"
"${_pkgname}-${pkgver}-rocm.zip::https://download.pytorch.org/libtorch/rocm5.7/libtorch-cxx11-abi-shared-with-deps-${pkgver}%2B${_rocm_version}.zip"
"${_pkgname}-${pkgver}-cpu.zip::https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-${pkgver}%2Bcpu.zip"
)
noextract=(
"${_pkgname}-${pkgver}-cuda.zip"
"${_pkgname}-${pkgver}-rocm.zip"
"${_pkgname}-${pkgver}-cpu.zip"
)
sha256sums=(
"SKIP"
"SKIP"
"SKIP"
)
options=('!strip' '!debug' 'libtool' 'staticlibs')
package_libtorch-cxx11abi-cuda() {
pkgdesc="${_pkgdesc} (with CUDA support)"
provides=("libtorch-cxx11abi-cuda")
depends=(pybind11)
install -vdm755 "${pkgdir}/opt"
cd ${pkgdir}/opt
bsdtar -xv -f "${srcdir}/${_pkgname}-${pkgver}-cuda.zip"
rm -rf ${_pkgname}/include/pybind11
mv ${_pkgname} ${_pkgname}-cuda
}
package_libtorch-cxx11abi-rocm() {
pkgdesc="${_pkgdesc} (with ROCM support)"
provides=("libtorch-cxx11abi-rocm")
depends=(pybind11)
install -vdm755 "${pkgdir}/opt"
cd ${pkgdir}/opt
bsdtar -xv -f "${srcdir}/${_pkgname}-${pkgver}-rocm.zip"
rm -rf ${_pkgname}/include/pybind11
mv ${_pkgname} ${_pkgname}-rocm
}
package_libtorch-cxx11abi-cpu() {
pkgdesc="${_pkgdesc} (CPU only)"
provides=("libtorch-cxx11abi-cpu")
depends=(pybind11)
install -vdm755 "${pkgdir}/opt"
cd ${pkgdir}/opt
bsdtar -xv -f "${srcdir}/${_pkgname}-${pkgver}-cpu.zip"
rm -rf ${_pkgname}/include/pybind11
mv ${_pkgname} ${_pkgname}-cpu
}
|