From ab0cf0b02fb3a28c6b710b4be6b5beb61ae42895 Mon Sep 17 00:00:00 2001
From: 36000 <jk6.28@outlook.com>
Date: Thu, 29 Jan 2026 10:48:32 +0900
Subject: [PATCH] [ENH] require cuda version to be specified

---
 README.md      |  4 +++-
 pyproject.toml | 21 +++++++++++++++++----
 2 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 77415aa..9ae3163 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,9 @@
 # GPUStreamlines
 
 ## Installation
-To install, simply run `pip install .` in the top-level repository directory.
+To install from pypi, simply run `pip install "cuslines[cu13]"` or `pip install "cuslines[cu12]"` depending on your CUDA version.
+
+To install from dev, simply run `pip install ".[cu13]"` or `pip install ".[cu12]"` in the top-level repository directory.
 
 ## Running the examples
 This repository contains several example usage scripts.
diff --git a/pyproject.toml b/pyproject.toml
index bd5c323..bd789d8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,14 +13,27 @@ dependencies = [
     "nibabel",
     "tqdm",
     "dipy",
-    "trx-python",
+    "trx-python"
+]
+
+[project.optional-dependencies]
+cu13 = [
     "nvidia-cuda-runtime",
     "nvidia-curand-cu12",
-    "cuda-python",
-    "cuda-core",
-    "cuda-cccl"
+    "cuda-python<14",
+    "cuda-core[cu13]",
+    "cuda-cccl[cu13]"
+]
+
+cu12 = [
+    "nvidia-cuda-runtime-cu12",
+    "nvidia-curand-cu12",
+    "cuda-python<13",
+    "cuda-core[cu12]",
+    "cuda-cccl[cu12]"
 ]
 
+
 [tool.setuptools.packages.find]
 where = ["."]
 include = ["cuslines*"]