Add support for CUDA11

initze · Aug 31, 2024 · 1f38229 · 1f38229
1 parent 3da154c
commit 1f38229
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -62,7 +62,11 @@ train = "thaw_slump_segmentation.scripts.train:main"
 thaw-slump-segmentation = "thaw_slump_segmentation.main:cli"
 
 [project.optional-dependencies]
-cuda = [
+cuda11 = [
+    "cupy-cuda11x>=13.3.0",
+    "cucim-cu11>=24.8.0",
+]
+cuda12 = [
     "cupy-cuda12x>=13.3.0",
     "cucim-cu12==24.8.*",
 ]