Skip to content

Commit 6cf0b67

Browse files
author
Klaas Giesbertz
committed
Needed to find the cuda tollkit to get the proper environment variables set.
1 parent 0c6d7ec commit 6cf0b67

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

cmake/skala-torch.cmake

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,14 @@ if(NOT Torch_FOUND)
88
set(LIBTORCH_VERSION "2.9.1")
99
set(USE_CUDA_LIBTORCH FALSE) #default is not to use the cuda version but cpu version
1010
if(GAUXC_HAS_CUDA)
11+
find_package(CUDAToolkit)
1112
set(SUPPORTED_CUDA_VERSION_STRINGS "126" "128" "130")
1213
set(CUDA_VERSION_STRING ${CUDAToolkit_VERSION_MAJOR}${CUDAToolkit_VERSION_MINOR})
1314
if(CUDA_VERSION_STRING IN_LIST SUPPORTED_CUDA_VERSION_STRINGS)
1415
set(USE_CUDA_LIBTORCH TRUE)
1516
else()
16-
message(WARNING "CUDA toolkit version is ${CUDAToolkit_VERSION}, for which there is no libtorch to download.")
17-
message(WARNING "Falling back to cpu version of libtorch.")
17+
message(WARNING "CUDA toolkit version is ${CUDAToolkit_VERSION}, for which there is no libtorch to download.")
18+
message(WARNING "Falling back to cpu version of libtorch.")
1819
endif()
1920
endif()
2021

0 commit comments

Comments
 (0)