diff --git a/pyproject.toml b/pyproject.toml
index 30712e02..6957be8b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -56,6 +56,12 @@ priority = "primary"
 
 [[tool.poetry.source]]
 name = "pytorch"
+secondary = true
+url = "https://download.pytorch.org/whl/cu118"
+
+[[tool.poetry.source]]
+name = "pytorch-nightly"
+secondary = true
 url = "https://download.pytorch.org/whl/nightly/cu118"
 
 [[tool.poetry.source]]