This commit is contained in:
Haoming 2026-03-29 20:51:53 +07:00
parent d9f2e1e252
commit b152417393

View File

@ -97,7 +97,7 @@ def _torch_version() -> tuple[str, str]:
if m is None:
print("\n\nFailed to parse PyTorch version...")
ver = os.environ.get("PYTORCH_VERSION", "2.9.1+cu128")
ver = os.environ.get("PYTORCH_VERSION", "2.10.0+cu130")
print("Assuming: ", ver)
print('(you can change this with `export PYTORCH_VERSION="..."`)\n\n')
m = re.search(r"(\d+\.\d+\.\d+)(?:[^+]+)?\+(.+)", ver)