From 47b1ccb42f1b8b7b7f9e60de0df523c3c18f29c3 Mon Sep 17 00:00:00 2001
From: Jianqi Pan <jannchie@gmail.com>
Date: Fri, 21 Feb 2025 22:45:02 +0900
Subject: [PATCH] feat(dependency): add optional dependencies for cuda11 and
 cuda12

---
 pyproject.toml | 3 +++
 1 file changed, 3 insertions(+)

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