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

update dcu_version

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