Update setup.cfg to add cuda option

neuronets · Mar 23, 2024 · 624afec · 624afec
1 parent 902a124
commit 624afec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -51,6 +51,8 @@ console_scripts =
     nobrainer=nobrainer.cli.main:cli
 
 [options.extras_require]
+and-cuda =
+    tensorflow[and-cuda] >=2.13, <= 2.15.1
 dev =
     pre-commit
     pytest