|
|
@ -144,6 +144,7 @@ if (WITH_ANAKIN AND WITH_MKL) # only needed in CI
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (NOT ${CUDNN_VERSION} VERSION_LESS 7100)
|
|
|
|
if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
set(TRT_MODEL_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/trt")
|
|
|
|
set(TRT_MODEL_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/trt")
|
|
|
|
if (NOT EXISTS ${TRT_MODEL_INSTALL_DIR})
|
|
|
|
if (NOT EXISTS ${TRT_MODEL_INSTALL_DIR})
|
|
|
@ -153,3 +154,4 @@ if(WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
|
|
|
EXTRA_DEPS ${INFERENCE_EXTRA_DEPS}
|
|
|
|
ARGS --infer_model=${TRT_MODEL_INSTALL_DIR}/trt_test_models SERIAL)
|
|
|
|
ARGS --infer_model=${TRT_MODEL_INSTALL_DIR}/trt_test_models SERIAL)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
endif()
|
|
|
|