diff --git a/poetry.lock b/poetry.lock
index 0346137..4c0ee96 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2947,4 +2947,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<3.15"
-content-hash = "914eca918159def95f38d1366b6ba0a8146d7fa904b4df45becfbe5745ff8bc2"
+content-hash = "1a98a1675830c27077e77d0ad5f2eb0ef11091534bb7889d1fab6c6d4a6ffed6"
diff --git a/pyproject.toml b/pyproject.toml
index 8fbc4fa..945a83e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,11 @@ dependencies = [
 [tool.poetry]
 packages = [{ include = "src" }]
 
+[[tool.poetry.source]]
+name = "pytorch-cu124"
+url = "https://download.pytorch.org/whl/cu124"
+priority = "explicit"
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = ["test_*.py", "*_test.py"]