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

Use specific version of CUDA base on disto release

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

14
python-torch.spec
View file

@ -92,6 +92,15 @@
%bcond_with cuda
%endif
# Pick a version that works
%if %{with cuda}
%if 0%{?fedora} < 40
%global cuda_ver 12.4
%else
%global cuda_ver 12.5
%endif
%endif
# For testing compat-gcc
%global compat_gcc_major 13
%bcond_with compat_gcc
@ -850,8 +859,9 @@ export BUILD_CAFFE2=ON
%if %{with cuda}
%if %{without rocm}
export CUDACXX=/usr/local/cuda/bin/nvcc
export CPLUS_INCLUDE_PATH=/usr/local/cuda/include
export CPLUS_INCLUDE_PATH=/usr/local/cuda-%{cuda_ver}/include
export CUDACXX=/usr/local/cuda-%{cuda_ver}/bin/nvcc
export CUDA_HOME=/usr/local/cuda-%{cuda_ver}/
export USE_CUDA=ON
# The arches to build for
export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"