From 85674301fb4053bdb6f26bc5fc9a6759ce4309ac Mon Sep 17 00:00:00 2001 From: Limitex <76650151+Limitex@users.noreply.github.com> Date: Sun, 9 Nov 2025 01:45:59 +0900 Subject: [PATCH] chore: add nvidia source --- poetry.lock | 2 +- pyproject.toml | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/poetry.lock b/poetry.lock index 1d2d512..4bfc38b 100644 --- a/poetry.lock +++ b/poetry.lock @@ -3202,4 +3202,4 @@ type = ["pytest-mypy"] [metadata] lock-version = "2.1" python-versions = ">=3.10,<3.15" -content-hash = "9196350678b6ec36e59174de9adf77afce361fb0ce84efb5d1eeffebd9fbcbb7" +content-hash = "12b19e9873629aa7ba0b7cc3ec5a727740bc5d1bc09d75c9a64aa0e61f5e15f6" diff --git a/pyproject.toml b/pyproject.toml index f65e7b4..5cace7d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,6 +30,11 @@ name = "pytorch-cu124" url = "https://download.pytorch.org/whl/cu124" priority = "explicit" +[[tool.poetry.source]] +name = "nvidia" +url = "https://pypi.nvidia.com" +priority = "supplemental" + [tool.poetry.dependencies] torch = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" } torchaudio = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" }