Merge pull request #507 from Kitware/dev/darknet-docker-fix

add stubs directory to lib path
Kitware · Mar 11, 2019 · 03dec61 · 03dec61
2 parents 8d5c66d + 9b39b9b
commit 03dec61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMake/External_Darknet.cmake b/CMake/External_Darknet.cmake
@@ -31,7 +31,7 @@ else()
   unset(fletch_ENABLE_Darknet_CUDNN CACHE)
 endif()
 if(fletch_ENABLE_Darknet_CUDA)
-  set( DARKNET_GPU_ARGS -DUSE_GPU:BOOL=ON -DCUDA_TOOLKIT_ROOT_DIR:PATH=${CUDA_TOOLKIT_ROOT_DIR})
+  set( DARKNET_GPU_ARGS -DUSE_GPU:BOOL=ON -DCUDA_TOOLKIT_ROOT_DIR:PATH=${CUDA_TOOLKIT_ROOT_DIR} -DCMAKE_LIBRARY_PATH=${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs)
   if (fletch_ENABLE_Darknet_CUDNN)
     set(DARKNET_CUDNN_ARGS -DUSE_CUDNN:BOOL=ON -DCUDNN_ROOT_DIR:PATH=${CUDNN_ROOT_DIR})
   else()