@@ -206,7 +206,7 @@ else()
206206 "libmkl_vml_def.so.1"
207207 )
208208endif ()
209- set (OPENCV_LIBS
209+ set (TORCHVISION_LIBS
210210 $<IF:$<BOOL :${RHEL_BUILD} >,libjpeg.so.62,libjpeg.so>
211211 $<IF:$<BOOL :${RHEL_BUILD} >,libpng16.so.16,libpng16.so>
212212)
@@ -222,7 +222,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
222222 OUTPUT
223223 ${PT_LIBS}
224224 ${LIBTORCH_LIBS}
225- ${OPENCV_LIBS }
225+ ${TORCHVISION_LIBS }
226226 LICENSE.pytorch
227227 include /torch
228228 include /torchvision
@@ -264,7 +264,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
264264 COMMENT "Extracting pytorch and torchvision libraries and includes from ${TRITON_PYTORCH_DOCKER_IMAGE} "
265265 VERBATIM
266266 )
267- add_custom_target (ptlib_target DEPENDS ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS } )
267+ add_custom_target (ptlib_target DEPENDS ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS } )
268268 add_library (ptlib SHARED IMPORTED GLOBAL )
269269 add_dependencies (ptlib ptlib_target)
270270
@@ -430,7 +430,7 @@ install(
430430
431431if (${TRITON_PYTORCH_DOCKER_BUILD} )
432432 set (PT_LIB_PATHS "" )
433- FOREACH (plib ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS } )
433+ FOREACH (plib ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS } )
434434 set (PT_LIB_PATHS ${PT_LIB_PATHS} "${CMAKE_CURRENT_BINARY_DIR} /${plib} " )
435435 ENDFOREACH (plib)
436436
@@ -450,7 +450,7 @@ if (${TRITON_PYTORCH_DOCKER_BUILD})
450450 )
451451 endif () # TRITON_PYTORCH_ENABLE_TORCHTRT
452452
453- FOREACH (plib ${PT_LIBS} ${LIBTORCH_LIBS} ${OPENCV_LIBS } )
453+ FOREACH (plib ${PT_LIBS} ${LIBTORCH_LIBS} ${TORCHVISION_LIBS } )
454454 install (
455455 CODE
456456 "EXECUTE_PROCESS(
0 commit comments