parent
460ff4720d
commit
b0b19107ed
|
@ -96,3 +96,4 @@ bruai4j-native-common/cmake*
|
||||||
/cavis-dnn/cavis-dnn-core/build/resources/test/logback-test.xml
|
/cavis-dnn/cavis-dnn-core/build/resources/test/logback-test.xml
|
||||||
/cavis-dnn/cavis-dnn-core/build/test-results/cudaTest/TEST-org.deeplearning4j.gradientcheck.AttentionLayerTest.xml
|
/cavis-dnn/cavis-dnn-core/build/test-results/cudaTest/TEST-org.deeplearning4j.gradientcheck.AttentionLayerTest.xml
|
||||||
/cavis-dnn/cavis-dnn-core/build/tmp/jar/MANIFEST.MF
|
/cavis-dnn/cavis-dnn-core/build/tmp/jar/MANIFEST.MF
|
||||||
|
/.metadata/
|
||||||
|
|
|
@ -121,7 +121,7 @@ endfunction()
|
||||||
|
|
||||||
if (SD_CUDA)
|
if (SD_CUDA)
|
||||||
#enable_language(CUDA)
|
#enable_language(CUDA)
|
||||||
find_package(CUDAToolkit 11.4 REQUIRED)
|
find_package(CUDAToolkit 12.2 REQUIRED)
|
||||||
message(STATUS "CUDAToolkit_VERSION: ${CUDAToolkit_VERSION}")
|
message(STATUS "CUDAToolkit_VERSION: ${CUDAToolkit_VERSION}")
|
||||||
message(STATUS "CUDAToolkit_VERSION_MAJOR: ${CUDAToolkit_VERSION_MAJOR}")
|
message(STATUS "CUDAToolkit_VERSION_MAJOR: ${CUDAToolkit_VERSION_MAJOR}")
|
||||||
message(STATUS "CUDAToolkit_VERSION_MINOR: ${CUDAToolkit_VERSION_MINOR}")
|
message(STATUS "CUDAToolkit_VERSION_MINOR: ${CUDAToolkit_VERSION_MINOR}")
|
||||||
|
|
Loading…
Reference in New Issue