setup: search for ffmpeg in TORCHVISION_LIBRARY (#2612)

pytorch · Aug 24, 2020 · 739061b · 739061b
1 parent ba1b221
commit 739061b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -358,6 +358,7 @@ def get_extensions():
                     ffmpeg_include_dir,
                     extensions_dir,
                 ],
+                library_dirs=library_dirs,
                 libraries=[
                     'avcodec',
                     'avformat',