100 lines
2.7 KiB
Plaintext
100 lines
2.7 KiB
Plaintext
|
pkgbase = python-pytorch-rocm
|
||
|
|
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
|
||
|
|
pkgver = 1.7.1
|
||
|
|
pkgrel = 7
|
||
|
|
url = https://pytorch.org
|
||
|
|
arch = x86_64
|
||
|
|
license = BSD
|
||
|
|
makedepends = python
|
||
|
|
makedepends = python-setuptools
|
||
|
|
makedepends = python-yaml
|
||
|
|
makedepends = python-numpy
|
||
|
|
makedepends = cmake
|
||
|
|
makedepends = rocm
|
||
|
|
makedepends = rocm-libs
|
||
|
|
makedepends = miopen
|
||
|
|
makedepends = git
|
||
|
|
makedepends = ninja
|
||
|
|
makedepends = pkgconfig
|
||
|
|
makedepends = doxygen
|
||
|
|
depends = google-glog
|
||
|
|
depends = gflags
|
||
|
|
depends = opencv
|
||
|
|
depends = openmp
|
||
|
|
depends = rccl
|
||
|
|
depends = pybind11
|
||
|
|
depends = python
|
||
|
|
depends = python-yaml
|
||
|
|
depends = libuv
|
||
|
|
depends = python-numpy
|
||
|
|
depends = protobuf
|
||
|
|
depends = ffmpeg
|
||
|
|
depends = python-future
|
||
|
|
depends = qt5-base
|
||
|
|
depends = onednn
|
||
|
|
depends = intel-mkl
|
||
|
|
source = pytorch-1.7.1::git+https://github.com/pytorch/pytorch.git#tag=v1.7.1
|
||
|
|
source = fix_include_system.patch
|
||
|
|
source = use-system-libuv.patch
|
||
|
|
source = use-system-libuv2.patch
|
||
|
|
source = nccl_version.patch
|
||
|
|
source = disable_non_x86_64.patch
|
||
|
|
source = find-hsa-runtime.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/45550.patch
|
||
|
|
sha256sums = SKIP
|
||
|
|
sha256sums = 83c81ec6a461110da6ae6182529f58100986b068c5182ca62cd53c648b4e4fb0
|
||
|
|
sha256sums = 26b1dd596f1e21a011ee18cab939924483d6c6d4d98e543bf76f5a9312d54d67
|
||
|
|
sha256sums = 7b65c3b209fc39f92ba58a58be6d3da40799f1922910b1171ccd9209eda1f9eb
|
||
|
|
sha256sums = e4a96887b41cbdfd4204ce5f16fcb16a23558d23126331794ab6aa30a66f2e0d
|
||
|
|
sha256sums = d3ef8491718ed7e814fe63e81df2f49862fffbea891d2babbcb464796a1bd680
|
||
|
|
sha256sums = SKIP
|
||
|
|
|
||
|
|
pkgname = python-pytorch-rocm
|
||
|
|
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM)
|
||
|
|
depends = google-glog
|
||
|
|
depends = gflags
|
||
|
|
depends = opencv
|
||
|
|
depends = openmp
|
||
|
|
depends = rccl
|
||
|
|
depends = pybind11
|
||
|
|
depends = python
|
||
|
|
depends = python-yaml
|
||
|
|
depends = libuv
|
||
|
|
depends = python-numpy
|
||
|
|
depends = protobuf
|
||
|
|
depends = ffmpeg
|
||
|
|
depends = python-future
|
||
|
|
depends = qt5-base
|
||
|
|
depends = onednn
|
||
|
|
depends = intel-mkl
|
||
|
|
depends = rocm
|
||
|
|
depends = rocm-libs
|
||
|
|
depends = miopen
|
||
|
|
provides = python-pytorch
|
||
|
|
conflicts = python-pytorch
|
||
|
|
|
||
|
|
pkgname = python-pytorch-opt-rocm
|
||
|
|
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM and AVX2 CPU optimizations)
|
||
|
|
depends = google-glog
|
||
|
|
depends = gflags
|
||
|
|
depends = opencv
|
||
|
|
depends = openmp
|
||
|
|
depends = rccl
|
||
|
|
depends = pybind11
|
||
|
|
depends = python
|
||
|
|
depends = python-yaml
|
||
|
|
depends = libuv
|
||
|
|
depends = python-numpy
|
||
|
|
depends = protobuf
|
||
|
|
depends = ffmpeg
|
||
|
|
depends = python-future
|
||
|
|
depends = qt5-base
|
||
|
|
depends = onednn
|
||
|
|
depends = intel-mkl
|
||
|
|
depends = rocm
|
||
|
|
depends = rocm-libs
|
||
|
|
depends = miopen
|
||
|
|
provides = python-pytorch
|
||
|
|
provides = python-pytorch-rocm
|
||
|
|
conflicts = python-pytorch
|
||
|
|
|