!19022 fix opencv install on python3.9

Merge pull request !19022 from zhoufeng/fix-py39-opencv-install
yxyOo · Jun 28, 2021 · 627b18b · 627b18b
2 parents 3629df5 + 419da71
commit 627b18b
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/cmake/package.cmake b/cmake/package.cmake
@@ -99,13 +99,21 @@ if(ENABLE_MINDDATA)
             COMPONENT mindspore
         )
     endif()
-    install(FILES ${opencv_LIBPATH}/libopencv_core.so.4.2.0
-      DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_core.so.4.2 COMPONENT mindspore)
-    install(FILES ${opencv_LIBPATH}/libopencv_imgcodecs.so.4.2.0
-      DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgcodecs.so.4.2 COMPONENT mindspore)
-    install(FILES ${opencv_LIBPATH}/libopencv_imgproc.so.4.2.0
-      DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgproc.so.4.2 COMPONENT mindspore)
-
+    if(PYTHON_VERSION MATCHES "3.8" OR PYTHON_VERSION MATCHES "3.7")
+      install(FILES ${opencv_LIBPATH}/libopencv_core.so.4.2.0
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_core.so.4.2 COMPONENT mindspore)
+      install(FILES ${opencv_LIBPATH}/libopencv_imgcodecs.so.4.2.0
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgcodecs.so.4.2 COMPONENT mindspore)
+      install(FILES ${opencv_LIBPATH}/libopencv_imgproc.so.4.2.0
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgproc.so.4.2 COMPONENT mindspore)
+    elseif(PYTHON_VERSION MATCHES "3.9")
+      install(FILES ${opencv_LIBPATH}/libopencv_core.so.4.5.1
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_core.so.4.5 COMPONENT mindspore)
+      install(FILES ${opencv_LIBPATH}/libopencv_imgcodecs.so.4.5.1
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgcodecs.so.4.5 COMPONENT mindspore)
+      install(FILES ${opencv_LIBPATH}/libopencv_imgproc.so.4.5.1
+        DESTINATION ${INSTALL_LIB_DIR} RENAME libopencv_imgproc.so.4.5 COMPONENT mindspore)
+    endif()
     install(FILES ${tinyxml2_LIBPATH}/libtinyxml2.so.8.0.0
       DESTINATION ${INSTALL_LIB_DIR} RENAME libtinyxml2.so.8 COMPONENT mindspore)