Replace pybind11 with cmake's Python3 package (#1850)

CMakeLists.txt

@@ -10,8 +10,8 @@ if(WITH_CUDA)
   add_definitions(-D__CUDA_NO_HALF_OPERATORS__)
 endif()
 
+find_package(Python3 COMPONENTS Development)
 find_package(Torch REQUIRED)
-find_package(pybind11 REQUIRED)
 
 file(GLOB HEADERS torchvision/csrc/*.h)
 file(GLOB OPERATOR_SOURCES torchvision/csrc/cpu/*.h torchvision/csrc/cpu/*.cpp)
@@ -22,7 +22,7 @@ file(GLOB MODELS_HEADERS torchvision/csrc/models/*.h)
 file(GLOB MODELS_SOURCES torchvision/csrc/models/*.h torchvision/csrc/models/*.cpp)
 
 add_library(${PROJECT_NAME} SHARED ${MODELS_SOURCES} ${OPERATOR_SOURCES})
-target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} pybind11::pybind11)
+target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} Python3::Python)
 set_target_properties(${PROJECT_NAME} PROPERTIES EXPORT_NAME TorchVision)
 
 target_include_directories(${PROJECT_NAME} INTERFACE

cmake/TorchVisionConfig.cmake.in

@@ -28,10 +28,10 @@ include("${CMAKE_CURRENT_LIST_DIR}/${PN}Targets.cmake")
 if(NOT TARGET torch_library)
 find_package(Torch REQUIRED)
 endif()
-if(NOT TARGET pybind11::pybind11)
-find_package(pybind11 REQUIRED)
+if(NOT TARGET Python3::Python)
+find_package(Python3 COMPONENTS Development)
 endif()
-target_link_libraries(TorchVision::TorchVision INTERFACE ${TORCH_LIBRARIES} pybind11::pybind11)
+target_link_libraries(TorchVision::TorchVision INTERFACE ${TORCH_LIBRARIES} Python3::Python)
 
 if(@WITH_CUDA@)
   target_compile_definitions(TorchVision::TorchVision INTERFACE WITH_CUDA)