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

Add CUDA BuildRequires

Browse source 
Signed-off-by: Tom Rix <trix@redhat.com>
This commit is contained in:
Tom Rix 2024-06-25 08:09:08 -04:00
commit bcdd1822e3
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
python-torch.spec
View file

@ -100,7 +100,12 @@
# Pick a version that works
%if %{with cuda}
%if 0%{?fedora} < 40
%global cuda_ver 12.4
%global cuda_ver 12.5
%global cudart_ver 12-5
%global cublas_ver 12-5
%global cufft_ver 12-5
%global curand_ver 12-5
%global cusparse_ver 12-5
%else
%global cuda_ver 12.5
%endif
@ -378,6 +383,14 @@ BuildRequires: roctracer-devel
Requires: rocm-rpm-macros-modules
%endif
%if %{with cuda}
BuildRequires: cuda-cudart-devel-%{cudart_ver}
BuildRequires: libcublas-devel-%{cublas_ver}
BuildRequires: libcufft-devel-%{cufft_ver}
BuildRequires: libcurand-devel-%{curand_ver}
BuildRequires: libcusparse-devel-%{cusparse_ver}
%endif
%if %{with opencv}
BuildRequires: opencv-devel
%endif