saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

onnxruntime_CUDA_HOME should only be used when available.

Browse source
This commit is contained in:
Tongliang Liao 2018-12-16 08:19:31 +08:00 committed by Changming Sun
parent 6cd935e184
commit 16b3a9fd63
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cmake/CMakeLists.txt
View file

@ -437,7 +437,9 @@ if (onnxruntime_USE_CUDA)
add_definitions(-DUSE_CUDA=1)
enable_language(CUDA)
set(CMAKE_CUDA_STANDARD 14)
file(TO_CMAKE_PATH ${onnxruntime_CUDA_HOME} CUDA_TOOLKIT_ROOT_DIR)
if (DEFINED onnxruntime_CUDA_HOME)
file(TO_CMAKE_PATH ${onnxruntime_CUDA_HOME} CUDA_TOOLKIT_ROOT_DIR)
endif ()
file(TO_CMAKE_PATH ${onnxruntime_CUDNN_HOME} onnxruntime_CUDNN_HOME)
find_package(CUDA 9.0 REQUIRED)
include(cub)
@ -498,4 +500,4 @@ if (onnxruntime_BUILD_UNIT_TESTS)
# turned ON by the previous step when building a shared lib
set(CMAKE_SKIP_BUILD_RPATH OFF)
include(onnxruntime_unittests.cmake)
endif()
endif()