Name:           torchvision
Version:        0.15.2
Release:        1%{?dist}
Summary:        Torchvision package
License:        BSD
URL:            https://github.com/pytorch/vision
Source0:        https://openi.pcl.ac.cn/JunJun-Liu/torchvision/vision-0.15.2.tar.gz

BuildRequires:  python3-devel
BuildRequires:  python3-pytorch >= 2.0.1
BuildRequires:  gcc
BuildRequires:  make
BuildRequires:  ninja-build
Requires:       python3
Requires:       python3-pytorch >= 2.0.1
Requires:       python3-numpy
Requires:       python3-pillow
Requires:       python3-scipy


%description
Torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.

%prep
%autosetup -n vision-0.15.2

%build
echo "Starting to build, Python site-lib path: %{python3_sitelib}"
python3 setup.py build

%install
echo "Starting to install, Python site-lib path: %{python3_sitearch}"
python3 setup.py install --root %{buildroot} --skip-build
find %{buildroot}%{python3_sitearch}/torchvision -type f

%files
/usr/lib64/python3.9/site-packages/torchvision*


%changelog
* Wed Feb 28 2024 Junjun Liu <junjunliu@example.com>  - 0.15.2-1
- Initial package.
- Update to support PyTorch 2.0.1.
- Add ninja-build to BuildRequires to support the build process.