Skip to content

GitLab

Unverified Commit a9a8220e authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Print FFmpeg location in setup.py (#2734)

parent a98e17e5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
setup.py
View file @ a9a8220e
...@@ -331,11 +331,13 @@ def get_extensions(): ...@@ -331,11 +331,13 @@ def get_extensions():
ffmpeg_exe = distutils.spawn.find_executable('ffmpeg') ffmpeg_exe = distutils.spawn.find_executable('ffmpeg')
has_ffmpeg = ffmpeg_exe is not None has_ffmpeg = ffmpeg_exe is not None
print("FFmpeg found: {}".format(has_ffmpeg))
if has_ffmpeg: if has_ffmpeg:
ffmpeg_bin = os.path.dirname(ffmpeg_exe) ffmpeg_bin = os.path.dirname(ffmpeg_exe)
ffmpeg_root = os.path.dirname(ffmpeg_bin) ffmpeg_root = os.path.dirname(ffmpeg_bin)
ffmpeg_include_dir = os.path.join(ffmpeg_root, 'include') ffmpeg_include_dir = os.path.join(ffmpeg_root, 'include')
print("ffmpeg include path: {}".format(ffmpeg_include_dir))
# TorchVision base decoder + video reader # TorchVision base decoder + video reader
video_reader_src_dir = os.path.join(this_dir, 'torchvision', 'csrc', 'cpu', 'video_reader') video_reader_src_dir = os.path.join(this_dir, 'torchvision', 'csrc', 'cpu', 'video_reader')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment