Merge pull request #33 from 36000/easier_cu12_support

[ENH] require cuda version to be specified

Author: 36000

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.

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*"]