mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
42a4df9447
Add a new option `--cuda` to `tools/nightly.py` to pull the nightly packages with CUDA support. ```bash # installs pytorch-nightly with cpuonly tools/nightly.py pull # The following only available on Linux and Windows # installs pytorch-nightly with latest CUDA we support tools/nightly.py pull --cuda # installs pytorch-nightly with CUDA 12.1 tools/nightly.py pull --cuda 12.1 ``` Also add targets in `Makefile` and instructions in constribution guidelines. ```bash # setup conda environment with pytorch-nightly make setup-env # setup conda environment with pytorch-nightly with CUDA support make setup-env-cuda ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/131133 Approved by: https://github.com/ezyang
83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
from enum import Enum
|
|
from typing import NamedTuple
|
|
|
|
|
|
LINTER_CODE = "LINTRUNNER_VERSION"
|
|
|
|
|
|
class LintSeverity(str, Enum):
|
|
ERROR = "error"
|
|
WARNING = "warning"
|
|
ADVICE = "advice"
|
|
DISABLED = "disabled"
|
|
|
|
|
|
class LintMessage(NamedTuple):
|
|
path: str | None
|
|
line: int | None
|
|
char: int | None
|
|
code: str
|
|
severity: LintSeverity
|
|
name: str
|
|
original: str | None
|
|
replacement: str | None
|
|
description: str | None
|
|
|
|
|
|
def toVersionString(version_tuple: tuple[int, int, int]) -> str:
|
|
return ".".join(str(x) for x in version_tuple)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
version_str = (
|
|
subprocess.run(["lintrunner", "-V"], stdout=subprocess.PIPE)
|
|
.stdout.decode("utf-8")
|
|
.strip()
|
|
)
|
|
|
|
import re
|
|
|
|
version_match = re.compile(r"lintrunner (\d+)\.(\d+)\.(\d+)").match(version_str)
|
|
|
|
if not version_match:
|
|
err_msg = LintMessage(
|
|
path="<none>",
|
|
line=None,
|
|
char=None,
|
|
code=LINTER_CODE,
|
|
severity=LintSeverity.ERROR,
|
|
name="command-failed",
|
|
original=None,
|
|
replacement=None,
|
|
description="Lintrunner is not installed, did you forget to run `make setup-lint && make lint`?",
|
|
)
|
|
sys.exit(0)
|
|
|
|
curr_version = int(version_match[1]), int(version_match[2]), int(version_match[3])
|
|
min_version = (0, 10, 7)
|
|
|
|
if curr_version < min_version:
|
|
err_msg = LintMessage(
|
|
path="<none>",
|
|
line=None,
|
|
char=None,
|
|
code=LINTER_CODE,
|
|
severity=LintSeverity.ADVICE,
|
|
name="command-failed",
|
|
original=None,
|
|
replacement=None,
|
|
description="".join(
|
|
(
|
|
f"Lintrunner is out of date (you have v{toVersionString(curr_version)} ",
|
|
f"instead of v{toVersionString(min_version)}). ",
|
|
"Please run `pip install lintrunner -U` to update it",
|
|
)
|
|
),
|
|
)
|
|
print(json.dumps(err_msg._asdict()), flush=True)
|