diff --git a/pyproject.toml b/pyproject.toml
index 2840135..649ee14 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,6 +14,9 @@ dev = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "ruff>=0.8.0"]
 [project.optional-dependencies]
 cpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
 gpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
+cuda11 = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
+cuda12 = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
+
 
 [tool.ruff]
 line-length = 140