Ship libpng and zlib with package on linux

Author: r-zenine

setup.py

@@ -154,11 +154,12 @@ def get_extensions():
         extension(
             'torchvision._C',
             sources,
-            libraries= libraries,
+            libraries=libraries,
             library_dirs=third_party_search_directories,
             include_dirs=include_dirs + third_party_search_directories,
             define_macros=define_macros,
             extra_compile_args=extra_compile_args,
+            runtime_library_dirs=['lib']
         )
     ]
     if compile_cpp_tests:
@@ -210,6 +211,7 @@ def run(self):
         # It's an old-style class in Python 2.7...
         distutils.command.clean.clean.run(self)
 
+
 def throw_of_failure(command):
     ret = os.system(command)
     if ret != 0:
@@ -222,22 +224,31 @@ def build_deps():
         cpu_count = multiprocessing.cpu_count()
         os.chdir("third_party/zlib/")
         throw_of_failure('cmake .')
-        throw_of_failure("cmake --build . --parallel {}".format(cpu_count))
+        throw_of_failure("cmake --build . -- -j {}".format(cpu_count))
         os.chdir(this_dir)
 
         zlib_path = os.path.join(this_dir, "third_party/zlib")
-        libpng_cmake_options="-DPNG_BUILD_ZLIB=ON -DPNG_STATIC=OFF -DZLIB_INCLUDE_DIR:PATH={zlib_path} -DZLIB_LIBRARY:FILEPATH={zlib_path}/libz.so".format(zlib_path=zlib_path)
+        libpng_cmake_options = "-DPNG_BUILD_ZLIB=ON -DPNG_STATIC=OFF -DZLIB_INCLUDE_DIR:PATH={zlib_path} -DZLIB_LIBRARY:FILEPATH={zlib_path}/libz.so".format(zlib_path=zlib_path)
         os.chdir("third_party/libpng/")
         os.system('cmake {} .'.format(libpng_cmake_options))
-        throw_of_failure("cmake --build . --parallel {}".format(cpu_count))
+        throw_of_failure("cmake --build . -- -j {}".format(cpu_count))
         os.chdir(this_dir)
 
 
-
 def build_ext_with_dependencies(self):
     build_deps()
     return BuildExtension.with_options(no_python_abi_suffix=True)(self)
 
+
+data_files = []
+if sys.platform.startswith('linux'):
+    data_files = [
+        ('torchvision/lib', [
+            'third_party/zlib/libz.so',
+            'third_party/libpng/libpng.so'])
+    ]
+
+
 setup(
     # Metadata
     name=package_name,
@@ -261,5 +272,6 @@ def build_ext_with_dependencies(self):
     cmdclass={
         'build_ext': build_ext_with_dependencies,
         'clean': clean,
-    }
+    },
+    data_files=data_files
 )