summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCoprDistGit <infra@openeuler.org>2024-04-15 15:27:51 +0000
committerCoprDistGit <infra@openeuler.org>2024-04-15 15:27:51 +0000
commita800df8592f89027932600c0e2abc6311e64d588 (patch)
tree181b1377a47a08923e345ca053570d23b0eff23f
parenta0305c67fa53e76cbcba89a2fbb68af5341feb1a (diff)
automatic import of torchvision
-rw-r--r--torchvision.spec24
1 files changed, 20 insertions, 4 deletions
diff --git a/torchvision.spec b/torchvision.spec
index b279a4e..863d46e 100644
--- a/torchvision.spec
+++ b/torchvision.spec
@@ -4,11 +4,27 @@ Version: 0.16.2
Release: 1
License: BSD
Group: Development/Libraries
-Source0: https://github.com/pytorch/vision/archive/refs/tags/v0.16.2.tar.gz
-Source1: tst.patch
+Source: https://github.com/pytorch/vision/archive/refs/tags/v0.16.2.tar.gz
URL: https://github.com/pytorch/vision
BuildRoot: %{_tmppath}/%{name}-buildroot
+Patch0: tst.patch
+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
@@ -23,7 +39,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.)
@@ -33,7 +49,7 @@ Torchvision is a PyTorch add-on that provides access to:
%prep
%autosetup -p1 -n vision-%{pypi_version}
-patch -p1 < %{SOURCE1}
+%patch0 -p1
%build
%pyproject_build