From 1311b02e130c02fdeab8faf67852f4a1e8ac2b2d Mon Sep 17 00:00:00 2001
From: Tom Rix <trix@redhat.com>
Date: Thu, 13 Jun 2024 09:35:46 -0400
Subject: [PATCH] Use specific version of CUDA base on disto release

Signed-off-by: Tom Rix <trix@redhat.com>
---
 python-torch.spec | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/python-torch.spec b/python-torch.spec
index 29db936..06fe5c1 100644
--- a/python-torch.spec
+++ b/python-torch.spec
@@ -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"