%FILENAME%
python-pytorch-cuda-2.2.2-1-aarch64.pkg.tar.zst

%NAME%
python-pytorch-cuda

%BASE%
python-pytorch

%VERSION%
2.2.2-1

%DESC%
Tensors and Dynamic neural networks in Python with strong GPU acceleration (with CUDA)

%CSIZE%
502579612

%ISIZE%
2027439774

%MD5SUM%
f1c401a0bf505028147a2bdd9c827402

%SHA256SUM%
66e22da86cba188f985fd270989c1e3c184be650b9ecbd7eebd31f77d666ad3e

%URL%
https://pytorch.org

%LICENSE%
BSD

%ARCH%
aarch64

%BUILDDATE%
1750249393

%PACKAGER%
GitLab CI Package Builder <ci-package-builder-1@steamos.cloud>

%CONFLICTS%
python-pytorch

%PROVIDES%
python-pytorch=2.2.2

%DEPENDS%
google-glog
gflags
opencv
openmp
openmpi
pybind11
python
python-yaml
libuv
python-numpy
python-sympy
protobuf
ffmpeg
python-future
qt6-base
intel-oneapi-mkl
python-typing_extensions
numactl
python-jinja
python-networkx
python-filelock
cuda
nccl
cudnn
magma-cuda
onednn

%MAKEDEPENDS%
python
python-setuptools
python-yaml
python-numpy
cmake
cuda
nccl
cudnn
git
rocm-hip-sdk
roctracer
miopen
magma-cuda
magma-hip
ninja
pkgconfig
doxygen
vulkan-headers
shaderc
onednn

