rpms/python-torch
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add a CUDA subpackage

Browse source 
Signed-off-by: Tom Rix <trix@redhat.com>
This commit is contained in:
Tom Rix 2024-06-26 09:46:41 -04:00
commit 3f2aad9a21
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
python-torch.spec
View file

@ -467,6 +467,14 @@ PyTorch is a Python package that provides two high-level features:
You can reuse your favorite Python packages such as NumPy, SciPy,
and Cython to extend PyTorch when needed.
%if %{with cuda}
%package -n python3-%{pypi_name}-cuda-%{cudart_ver}
Summary: %{name} for CUDA
%description -n python3-%{pypi_name}-cuda-%{cudart_ver}
%{summary}
%endif
%if %{with rocm}
%package -n python3-%{pypi_name}-rocm-gfx8
Summary: %{name} for ROCm gfx8
@ -1029,8 +1037,11 @@ done
# Do not remote the empty files
%files -n python3-%{pypi_name}
%if %{with cuda}
%files -n python3-%{pypi_name}-cuda-%{cudart_ver}
%else
%files -n python3-%{pypi_name}
%endif
%license LICENSE
%doc README.md
%{_bindir}/convert-caffe2-to-onnx