Update Caffe2 CMake CUDA flag

Reviewed By: rbgirshick

Differential Revision: D8541427

fbshipit-source-id: f681d75ad77ea0c7411e5c6099c6bb0c2497d571

CMakeLists.txt

@@ -38,7 +38,7 @@ target_link_libraries(caffe2_detectron_custom_ops caffe2_library)
 install(TARGETS caffe2_detectron_custom_ops DESTINATION lib)
 
 # Install custom GPU ops lib, if gpu is present.
-if (${CAFFE2_FOUND_CUDA})
+if ("${CAFFE2_USE_CUDA}" OR "${CAFFE2_FOUND_CUDA}")
   # Additional -I prefix is required for CMake versions before commit (< 3.7):
   # https://github.com/Kitware/CMake/commit/7ded655f7ba82ea72a82d0555449f2df5ef38594
   list(APPEND CUDA_INCLUDE_DIRS -I${CAFFE2_INCLUDE_DIRS})

cmake/Summary.cmake

@@ -12,7 +12,7 @@ function (detectron_print_config_summary)
   message(STATUS "  Caffe2 version       : ${CAFFE2_VERSION}")
   message(STATUS "  Caffe2 include path  : ${CAFFE2_INCLUDE_DIRS}")
   message(STATUS "  Caffe2 found CUDA    : ${CAFFE2_FOUND_CUDA}")
-  if (${CAFFE2_FOUND_CUDA})
+  if ("${CAFFE2_USE_CUDA}" OR "${CAFFE2_FOUND_CUDA}")
     message(STATUS "    CUDA version       : ${CUDA_VERSION}")
     message(STATUS "    CuDNN version      : ${CUDNN_VERSION}")
   endif()