chore: add nvidia source

This commit is contained in:
Limitex
2025-11-12 23:21:46 +09:00
parent 0d1de5af4c
commit 85674301fb
2 changed files with 6 additions and 1 deletions
Generated
+1 -1
View File
@@ -3202,4 +3202,4 @@ type = ["pytest-mypy"]
[metadata]
lock-version = "2.1"
python-versions = ">=3.10,<3.15"
content-hash = "9196350678b6ec36e59174de9adf77afce361fb0ce84efb5d1eeffebd9fbcbb7"
content-hash = "12b19e9873629aa7ba0b7cc3ec5a727740bc5d1bc09d75c9a64aa0e61f5e15f6"
+5
View File
@@ -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" }