Commit 00670edc authored by panning's avatar panning
Browse files

update dcu_version

parent 5343fb43
......@@ -74,7 +74,7 @@ def write_version_file():
version_path = os.path.join(cwd, "torchvision", "version.py")
with open(version_path, "w") as f:
f.write(f"__version__ = '{version}'\n")
f.write("__dcu_version__ = '{}'\n".format(dcu_version))
f.write(f"__dcu_version__ = '{dcu_version}'\n")
f.write(f"git_version = {repr(sha)}\n")
f.write("from torchvision.extension import _check_cuda_version\n")
f.write("if _check_cuda_version() > 0:\n")
......@@ -552,7 +552,7 @@ if __name__ == "__main__":
write_version_file()
with open("README.rst") as f:
with open("README_ORIGIN.rst") as f:
readme = f.read()
setup(
......
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