diff --git a/tools/generate_torch_version.py b/tools/generate_torch_version.py index 61682c9c896344..2ee17b76e52f4a 100644 --- a/tools/generate_torch_version.py +++ b/tools/generate_torch_version.py @@ -13,7 +13,7 @@ def get_sha(pytorch_root: Union[str, Path]) -> str: def get_torch_version(sha: Optional[str] = None) -> str: pytorch_root = Path(__file__).parent.parent - version = open('version.txt', 'r').read().strip() + version = open(pytorch_root / 'version.txt', 'r').read().strip() if os.getenv('PYTORCH_BUILD_VERSION'): assert os.getenv('PYTORCH_BUILD_NUMBER') is not None