Add extra build flag

VIAME · Aug 2, 2023 · f80eec8 · f80eec8
1 parent 8e48bae
commit f80eec8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmake/add_project_pytorch.cmake b/cmake/add_project_pytorch.cmake
@@ -94,6 +94,10 @@ if( VIAME_ENABLE_PYTORCH-MMDET )
   list( APPEND PYTORCH_ENV_VARS "MMCV_WITH_OPS=1" )
 endif()
 
+if( WIN32 AND VIAME_ENABLE_LEARN AND Python_VERSION VERSION_GREATER "3.7" )
+  list( APPEND PYTORCH_ENV_VARS "SETUPTOOLS_USE_DISTUTILS=1" )
+endif()
+
 foreach( LIB ${PYTORCH_LIBS_TO_BUILD} )
   if( "${LIB}" STREQUAL "pytorch" )
     set( LIBRARY_LOCATION ${VIAME_PACKAGES_DIR}/pytorch )