diff --git a/setup.cfg b/setup.cfg index a8277e0c..7faa9916 100644 --- 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