Name:           pytorch-lightning
Version:        2.2.0
Release:        1%{?dist}
Summary:        The lightweight PyTorch wrapper for high-performance AI research

License:        Apache-2.0
URL:            https://github.com/PyTorchLightning/pytorch-lightning
Source0:        https://openi.pcl.ac.cn/JunJun-Liu/pytorch-lightning/pytorch-lightning-%{version}.tar.gz

BuildRequires:  python3-devel
BuildRequires:  python3-setuptools
BuildRequires:  python3-pytorch >= 2.0.1
BuildRequires:  gcc
BuildRequires:  make
Requires:       python3
Requires:       python3-pytorch >= 2.0.1

%description
PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. 
Scale your models, not the boilerplate.

%prep
%autosetup -n pytorch-lightning-%{version}

%build
python3 setup.py build

%install
echo "Python site-lib directory: %{buildroot}%{python3_sitelib}"
python3 setup.py install --root=%{buildroot} --skip-build

%files -n pytorch-lightning
%license LICENSE
/usr/lib64/python3.9/site-packages/*

%changelog
* Wed Feb 28 2024 Junjun Liu - 2.2.0-1
- Initial package.