diff --git a/pyproject.toml b/pyproject.toml
index 20cdab9..128a5af 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -59,13 +59,18 @@ dev = [
 
 # Add PyTorch CUDA index as additional source
 # CUDA builds work on both GPU and CPU systems (fallback to CPU if no GPU)
+# Using 'explicit = true' means only packages we explicitly specify use this index
 [[tool.uv.index]]
 name = "pytorch-cu121"
 url = "https://download.pytorch.org/whl/cu121"
+explicit = true
 
-# Tell uv to get torch from the PyTorch CUDA index
+# Tell uv to get torch, torchvision, and torchaudio from the PyTorch CUDA index
+# All other packages come from PyPI
 [tool.uv.sources]
 torch = { index = "pytorch-cu121" }
+torchvision = { index = "pytorch-cu121" }
+torchaudio = { index = "pytorch-cu121" }
 
 [tool.ruff]
 line-length = 100