"vscode:/vscode.git/clone" did not exist on "36c87c433b7d880ef8b3a2b05ef93b0cd1675520"
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:
f.write("__version__ = '{}'\n".format(version))
f.write("git_version = {}\n".format(repr(sha)))
pytorch_package_name = os.getenv('TORCHAUDIO_PYTORCH_DEPENDENCY_NAME', 'torch')
setup(
name="torchaudio",
version="0.2",
......@@ -111,5 +113,5 @@ setup(
extra_link_args=ela),
],
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