Commit 9bc3cedb authored by Edward Z. Yang's avatar Edward Z. Yang Committed by Soumith Chintala
Browse files

Make it possible to change the name of torch (e.g., to torch_nightly) (#176)


Signed-off-by: default avatarEdward Z. Yang <ezyang@fb.com>
parent 95235f31
...@@ -76,6 +76,8 @@ with open(version_path, 'w') as f: ...@@ -76,6 +76,8 @@ with open(version_path, 'w') as f:
f.write("__version__ = '{}'\n".format(version)) f.write("__version__ = '{}'\n".format(version))
f.write("git_version = {}\n".format(repr(sha))) f.write("git_version = {}\n".format(repr(sha)))
pytorch_package_name = os.getenv('TORCHAUDIO_PYTORCH_DEPENDENCY_NAME', 'torch')
setup( setup(
name="torchaudio", name="torchaudio",
version="0.2", version="0.2",
...@@ -111,5 +113,5 @@ setup( ...@@ -111,5 +113,5 @@ setup(
extra_link_args=ela), extra_link_args=ela),
], ],
cmdclass={'build_ext': BuildExtension}, cmdclass={'build_ext': BuildExtension},
install_requires=['torch'] install_requires=[pytorch_package_name]
) )
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