chore: add ngc nvidia source

This commit is contained in:
Limitex
2025-11-12 23:27:21 +09:00
parent 563d5cf881
commit 4b0c19d34a
+5
View File
@@ -36,6 +36,11 @@ name = "nvidia"
url = "https://pypi.nvidia.com" url = "https://pypi.nvidia.com"
priority = "supplemental" priority = "supplemental"
[[tool.poetry.source]]
name = "ngc"
url = "https://pypi.ngc.nvidia.com"
priority = "supplemental"
[tool.poetry.dependencies] [tool.poetry.dependencies]
torch = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" } torch = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" }
torchaudio = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" } torchaudio = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" }