%global _empty_manifest_terminate_build 0
Name:           torchvision
Version:        0.17.1
Release:        1%{?dist}
Summary:        Image and video datasets and models for torch deep learning
License:        BSD-3-Clause
URL:            https://github.com/pytorch/vision
Source0:        https://openi.pcl.ac.cn/JunJun-Liu/torchvision-atom/vision-0.17.1.tar.gz
Source1:        https://openi.pcl.ac.cn/JunJun-Liu/torchvision-atom/pytorch-v2.0.1.tar.gz

BuildRequires:  gcc
BuildRequires:  python3-devel
BuildRequires:  python3-setuptools
BuildRequires:  python3-pip
BuildRequires:  cmake
BuildRequires:  python3-pyyaml
BuildRequires:  python3-typing-extensions
Requires:       python3-torch
Requires:       python3-numpy
Requires:       python3-pillow
Requires:       python3-scipy

%description
torchvision is a Python package that provides image and video datasets and models for torch deep learning applications. It includes utilities for efficient image and video transformations, common architectures for classification, segmentation, and detection, and pretrained models.

%prep
%autosetup -p1 -n vision-0.17.1
# Manually extract PyTorch source code for building
tar -xzf %{SOURCE1} -C .

%build
# Update PYTHONPATH for PyTorch and torchvision
export PYTHONPATH=/builddir/build/BUILD/vision-0.17.1/pytorch-v2.0.1/build/lib.linux-x86_64-3.9:$PYTHONPATH

# Build PyTorch from source
pushd pytorch-v2.0.1
python3 setup.py build
popd

# Update PYTHONPATH if necessary, ensure it points to where PyTorch was built
export PYTHONPATH=$(pwd)/pytorch-v2.0.1/build/lib:$PYTHONPATH

# Check if PyTorch is correctly detected
python3 -c "import torch; print(torch.__version__)"

# Build torchvision from source
echo "Starting build process in $(pwd)"
python3 setup.py build
echo "Build process completed"

%install
# Install PyTorch
pushd pytorch-v2.0.1
python3 setup.py install --skip-build --root %{buildroot}
popd

# Install torchvision
echo "Starting install process"
python3 setup.py install --skip-build --root %{buildroot}
echo "Install process completed"

mkdir -p %{buildroot}%{_pkgdocdir}
cp -arf docs %{buildroot}%{_pkgdocdir}
cp -arf examples %{buildroot}%{_pkgdocdir}
cp -arf LICENSE %{buildroot}%{_pkgdocdir}
cp -arf README*.md %{buildroot}%{_pkgdocdir}

%files
%doc %{_pkgdocdir}/README*.md
%license %{_pkgdocdir}/LICENSE
/usr/lib/python3.9/site-packages/torchvision*
%doc /usr/share/doc/torchvision/*

%changelog
* Tue Feb 28 2023 JunjunLiu <junjunliu@example.com> - 0.17.1-1
- First build of torchvision RPM