summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCoprDistGit <infra@openeuler.org>2024-04-15 15:12:21 +0000
committerCoprDistGit <infra@openeuler.org>2024-04-15 15:12:21 +0000
commit2aeb77007de43812cffab5131dd591ab53932ea3 (patch)
treed57a0ea485f934c58a29b14aa2e532a85b71c71f
parent3107e2bb9eed87f041fc862374bad14c8785051e (diff)
automatic import of torchvision
-rw-r--r--torchvision.spec18
1 files changed, 17 insertions, 1 deletions
diff --git a/torchvision.spec b/torchvision.spec
index c27e3df..a5b74d6 100644
--- a/torchvision.spec
+++ b/torchvision.spec
@@ -8,6 +8,22 @@ Source: https://github.com/pytorch/vision/archive/refs/tags/v0.16.2.tar.gz
URL: https://github.com/pytorch/vision
BuildRoot: %{_tmppath}/%{name}-buildroot
+Requires: python3-pytorch
+Requires: python3-numpy
+Requires: python3-pillow
+Requires: python3-requests
+Requires: python3-scipy
+Requires: python3-sympy
+
+%description
+Torchvision is a PyTorch add-on that provides access to:
+- Datasets (MNIST, CIFAR10, etc.)
+- Models (ResNet, VGG, etc.)
+- Transforms (for image preprocessing)
+
+%package -n python3-torchvision
+Summary: The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
+Provides: python-torchvision
BuildRequires: python3-pytorch
BuildRequires: python3-requests
BuildRequires: python3-pip
@@ -22,7 +38,7 @@ BuildRequires: ninja-build
BuildRequires: numactl
BuildRequires: cmake
-%description
+%description -n python3-torchvision
Torchvision is a PyTorch add-on that provides access to:
- Datasets (MNIST, CIFAR10, etc.)
- Models (ResNet, VGG, etc.)