diff --git a/pyproject.toml b/pyproject.toml
index c599887..7da7d2f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -60,6 +60,22 @@ dev = [
   "ruff>=0.7.3",
 ]
 
+[[tool.uv.index]]
+name = "pytorch-cu126"
+url = "https://download.pytorch.org/whl/cu126"
+explicit = true
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+torchvision = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+torchaudio = [
+  { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
 [tool.ruff]
 line-length = 79
 target-version = "py39"