Allow building without a python installation.

dfalbel · Jan 10, 2022 · 2a81375 · 2a81375
1 parent d675c0c
commit 2a81375
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -76,7 +76,7 @@ FOREACH(DIR ${ALLOW_LISTED})
 ENDFOREACH()
 
 add_library(${PROJECT_NAME} SHARED ${ALL_SOURCES})
-target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} ${PNG_LIBRARY} ${JPEG_LIBRARIES} Python3::Python)
+target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} ${PNG_LIBRARY} ${JPEG_LIBRARIES})
 set_target_properties(${PROJECT_NAME} PROPERTIES
   EXPORT_NAME TorchVision
   INSTALL_RPATH ${TORCH_INSTALL_PREFIX}/lib)

diff --git a/torchvision/csrc/io/image/image.cpp b/torchvision/csrc/io/image/image.cpp
@@ -1,7 +1,7 @@
 #include "image.h"
 
 #include <ATen/core/op_registration/op_registration.h>
-#include <Python.h>
+//#include <Python.h>
 
 // If we are in a Windows environment, we need to define
 // initialization functions for the _custom_ops extension

diff --git a/torchvision/csrc/io/video_reader/video_reader.cpp b/torchvision/csrc/io/video_reader/video_reader.cpp
@@ -1,6 +1,6 @@
 #include "video_reader.h"
 
-#include <Python.h>
+//#include <Python.h>
 
 #include "../decoder/memory_buffer.h"
 #include "../decoder/sync_decoder.h"

diff --git a/torchvision/csrc/vision.cpp b/torchvision/csrc/vision.cpp
@@ -1,7 +1,7 @@
 #include "vision.h"
 
 #ifndef MOBILE
-#include <Python.h>
+//#include <Python.h>
 #endif
 #include <torch/library.h>