From ec108a11398160e06a7d921206880093d0b46b11 Mon Sep 17 00:00:00 2001 From: moto <855818+mthrok@users.noreply.github.com> Date: Thu, 30 Jul 2020 17:01:40 +0000 Subject: [PATCH] Enable ABI suffix for extension module When installing torchaudio from source `python setup.py install` performs egg (zip) installation. This causes confusion when building from source for multiple environments with different Python versions, as `_torchaudio.so` does not have Python ABI version in its name and importlib cannot find it. This change fixs this by enableing ABI extension. squash! Enable ABI suffix for extension module --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index aab59b4d8a4..4cb9a2520f8 100644 --- a/setup.py +++ b/setup.py @@ -83,7 +83,7 @@ def run(self): packages=find_packages(exclude=["build*", "test*", "torchaudio.csrc*", "third_party*", "build_tools*"]), ext_modules=setup_helpers.get_ext_modules(), cmdclass={ - 'build_ext': setup_helpers.BuildExtension.with_options(no_python_abi_suffix=True) + 'build_ext': setup_helpers.BuildExtension, }, install_requires=[pytorch_package_dep] )